题解 | #牛群排队# Java

import java.util.*;


public class Solution {

    public int[][] cow_permute (int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        boolean[] used = new boolean[nums.length];
        Arrays.sort(nums);

        dfs(nums, used, new ArrayList<>(), res);
        int[][] ans = new int[res.size()][nums.length];
        for (int i = 0; i < res.size(); i++) {
            for (int j = 0; j < nums.length; j++) {
                ans[i][j] = res.get(i).get(j);
            }
        }

        return ans;
    }
    private void dfs(int[] nums, boolean[] used, List<Integer> path,
                     List<List<Integer>> res) {
        if (path.size() == nums.length) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = nums.length-1; i>=0 ; i--) {
            if (!used[i]) {
                path.add(nums[i]);
                used[i] = true;
                dfs(nums, used, path, res);
                used[i] = false;
                path.remove(path.size() - 1);
            }
        }
    }
}

dfs 方法是一个辅助方法，用于执行深度优先搜索。它接受四个参数：nums 表示牛的编号，used 是一个布尔数组，表示每头牛是否已经被使用，path 是一个列表，表示当前的排队情况，res 是一个列表，用于存储所有可能的排队情况。

在 dfs 方法中，首先检查 path 的大小是否等于 nums 的长度。如果是，则将 path 添加到 res 中。否则，从后往前遍历 nums 中的每个元素，如果该元素尚未被使用，则将其添加到 path 中，并将其标记为已使用。然后递归调用 dfs 函数继续搜索。最后，撤销对该元素的使用，并从 path 中删除该元素。