leetcode 77 组合-回溯加剪枝

给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

分析:此题为较简单,使用回溯加剪枝即可解决。
首先我们分析递归函数的参数,需要一个pos来表示当前选取了多少个数,同时还需要一个参数curr来表示当前是对1…n中的哪个数进行操作,对于每个curr都可以有选和不选两种选择,分别递归下去即可。
需要注意的是,这里有个剪枝小技巧,在剪枝之前耗时3ms,剪枝之后耗时1ms,看来剪枝的作用还是很大的。本题的剪枝方法是,如果当前已经选择的数的个数加上剩余最多可以选择的数的个数小于题目要求的个数k时,直接退出递归即可,因为根本无法选出k个数,这样剪枝会减少很多不必要的递归。
需要记住这个小技巧.

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

    }
    public void dfs(int curr,int pos,int k,int n){
        if(pos == k){
            ans.add(new ArrayList<Integer>(seq));
            return ;
        }else if(pos > k || curr > n){
            return;
        }else if(seq.size() + (n-curr + 1) <k){ //这里进行剪枝 如果已经选的长度加上剩余能够选的长度加起来
        //不可能构造出长度为k的temp,那么就直接退出
            return;
        }else {

            //选择当前数
            seq.add(curr);
            dfs(curr+1,pos + 1,k,n);

            //不选择当前数
            seq.remove(seq.size()-1);
            dfs(curr + 1,pos,k,n);
        }

    }
}

你可能感兴趣的:(leetcode刷题,剪枝,算法,java)