[算法专题 ] 组合

目录

需求：

        示例1

        示例2

思路：

 代码:

---

需求：

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

        示例1

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

        示例2

输入：n = 1, k = 1
输出：[[1]]

提示：

• 1 <= n <= 20
• 1 <= k <= n

思路：

 回溯函数传入n，k和选择的元素位置startIndex，在每层递归中，从startIndex开始循环到 n - (k - path.length) + 1的位置，将这些数加入path，然后startIndex加1，继续递归函数进入下一个分支，完成调用之后回溯状态，当path的长度等于k的时候终止这层分支，加入结果中。
复杂度：时间复杂度：O(C(n, k) * k)，枚举结果总数为C(n, k)，每次得到一个结果需要O(k)时间。空间复杂度：O(n)，最大是n层递归栈。

 代码:

class Method {
    List<List<Integer>> result = new ArrayList<>();
    LinkedList<Integer> path = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        combineHelper(n, k, 1);
        return result;
    }

    private void combineHelper(int n, int k, int startIndex){
        if (path.size() == k){
            result.add(new ArrayList<>(path));
            return;
        }
        for (int i = startIndex; i <= n - (k - path.size()) + 1; i++){
            path.add(i);
            combineHelper(n, k, i + 1);
            path.removeLast();
        }
    }
}

 zz1998