leetcode 77. 组合【回溯】

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

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