leetcode 77. 组合【回溯】

最新推荐文章于 2024-11-01 15:19:21 发布

BaldBear

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量146

点赞数

分类专栏：刷题路漫漫文章标签： leetcode java 回溯

本文链接：https://blog.csdn.net/er_ving/article/details/107233293

版权

刷题路漫漫专栏收录该内容

108 篇文章 1 订阅

订阅专栏

组合类的题目基本都是套用回溯的模板就可以解决。

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        helper(1,n, k, list);
        return res;
    }
    
    public void helper(int cur, int n, int k, List<Integer> list){
        if(list.size()==k){
            res.add(new ArrayList(list));
            return;
        }
        for(int i=cur; i<=n; i++){
            list.add(i);
            helper(i+1, n, k, list);
            list.remove(list.size()-1);
        }
    }
}

修改了for循环的终止条件，当list中的元素长度加上数字n-cur剩余的元素比k小时，无法构建出满足要求长度k的list，因此提前终止循环。

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        helper(1,n, k, list);
        return res;
    }
    
    public void helper(int cur, int n, int k, List<Integer> list){
        if(list.size()==k){
            res.add(new ArrayList(list));
            return;
        }
        for(int i=cur; i<=n-(k-list.size())+1; i++){
            list.add(i);
            helper(i+1, n, k, list);
            list.remove(list.size()-1);
        }
    }
}