0
点赞
收藏
分享

微信扫一扫

【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);
    }
}


举报

相关推荐

0 条评论