组合 - Java - 回溯 + 剪枝

一、77. 组合 - Java - 回溯 + 剪枝

1.1 题目描述

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

示例:

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

1.2 代码实现

public class BackTracking {
    /**给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合**/
    public static void main(String[] args) {
        List<List<Integer>> result = combine(4, 2);
        System.out.println(result.toString());
    }

    // 回溯法
    public static List<List<Integer>> combine(int n, int k) {
        if (n == 0 || k == 0) {
            return null;
        }
        // 返回的结果
        List<List<Integer>> 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<List<Integer>> res, List<Integer> cur) {
        // 递归结束条件
        if (cur.size() == k) {
            res.add(new ArrayList<>(cur));
            return;
        }

        // 从startIndex开始计算,
        /* 可以直接遍历到n ,但是没必要因为可以剪枝优化
           优化思路：
            结束时：搜索起点有上界，且搜索起点的上界 + 接下来要选择的元素个数 - 1 = n .
                接下来要选择的元素个数 = k - path.size()，整理得到：搜索起点的上界 = n - (k - path.size()) + 1
            如： n = 7, k = 4，从 5 开始搜索就已经没有意义了，这是因为：即使把 5 选上，
                后面的数只有 6 和 7，一共就 3 个候选数，凑不出 4 个数的组合
        */
        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);
        }
    }
}