【leetcode】含有 k 个元素的组合

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

java代码：

class Solution {

    List<List<Integer>> ans;

    public List<List<Integer>> combine(int n, int k) {
        ans = new ArrayList<>();
        dfs(n, 1, k, new ArrayList<>());
        return ans;
    }

    public void dfs(int n, int th, int k, List<Integer> list) {
        // 剪枝：即使把从th开始的所有数都放入list也凑不齐k个，所以直接返回
        if (n - th + 1 < k) return ;
        if (k == 0) {
            ans.add(new ArrayList<>(list));
            return ;
        }
        // 搜索策略一：组合中有第th个
        list.add(th);
        dfs(n, th + 1, k - 1, list);
        list.remove(list.size() - 1);
        // 搜索策略二：组合中没有第th个
        dfs(n, th + 1, k, list);
    }
}