组合 - Java - 回溯 + 剪枝

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

public class BackTracking {
    
    public static void main(String[] args) {
        List> result = combine(4, 2);
        System.out.println(result.toString());
    }

    // 回溯法
    public static List> combine(int n, int k) {
        if (n == 0 || k == 0) {
            return null;
        }
        // 返回的结果
        List> res = new ArrayList<>();
        // start从1开始，因为n 从1 开始的
        backtracking(n, k, 1, res, new ArrayList<>());
        return res;
    }

    // 回溯
    // 参数： k : 长度，  startIndex： 起始长度
    public static void backtracking(int n, int k, int startIndex, List> res, List cur) {
        // 递归结束条件
        if (cur.size() == k) {
            res.add(new ArrayList<>(cur));
            return;
        }

        // 从startIndex开始计算,
        
        for (int i = startIndex ; i <= n - (k - cur.size()) + 1 ; i++) {
            // 做选择
            cur.add(i);
            // 往下一层递归
            backtracking(n, k, i + 1, res, cur);
            // 取消选择
            cur.remove(cur.size() - 1);
        }
    }
}