Java 题解 | #牛棚分组#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param n int整型
     * @param k int整型
     * @return int整型二维数组
     */
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new ArrayList<>();
    boolean[] st = new boolean[20];

    public void dfs(int u, int n, int k) {
        if (path.size() == k) {
            res.add(new ArrayList<>(path));
            return;
        }

        for (int i = u; i <= n; i++) {
            if (!st[i]) {
                st[i] = true;
                path.add(i);
                dfs(i + 1, n, k);
                path.remove(path.size() - 1);
                st[i] = false;
            }
        }
    }

    public int[][] combine(int n, int k) {
        dfs(1, n, k);
        int[][] result = new int[res.size()][];
        for (int i = 0; i < res.size(); i++) {
            result[i] = new int[res.get(i).size()];
            for (int j = 0; j < res.get(i).size(); j++) {
                result[i][j] = res.get(i).get(j);
            }
        }
        return result;
    }
}

编程语言是Java。

该题考察的知识点是回溯算法。

代码的文字解释：

combine接受两个整数n和k作为输入，并返回一个二维整数数组作为输出。

在combine方法中，首先创建一个二维整数数组res和一个整数列表path来存储组合的结果和当前路径。同时创建一个布尔型数组st来记录数字是否被使用过。

使用dfs函数进行回溯，其中u表示当前考虑的数字，n表示范围，k表示组合长度。如果当前路径长度等于k，将路径加入结果数组res。

循环遍历从u到n的数字，如果数字没有被使用过，则将其标记为使用，将其加入路径，递归调用dfs函数进行下一层的搜索，然后将路径中的数字弹出，将其标记为未使用，以便进行下一次搜索。