LeetCode 刷题系列 -- 77. 组合

给定两个整数 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/combinations
 

思路：回溯算法秒杀所有排列/组合/子集问题 :: labuladong的算法小抄

回溯算法。
数量为 k 的组合等价于 数量为 k的子集

java：

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> result = new LinkedList<>();

        boolean[] used = new boolean[n+1]; // used[i] 用于判断 i 数字是否存在

        backtrack(n,k,used,1,result,new LinkedList<>());
        return result;
    }

    public void backtrack(int n, int k, boolean[] used, int start,List<List<Integer>> result, List<Integer> subRes) {

        if(k == subRes.size()) {
            result.add(new LinkedList<>(subRes));
            return;
        }

        for (int i = start; i <= n; i++) {
            if (used[i]) {
                continue;
            }
            used[i] = true;
            subRes.add(i);
            backtrack(n,k,used,i+1,result,subRes);
            used[i] = false;
            subRes.remove(subRes.size()-1);
        }
    }
}