77. 组合

77. 组合

进入回溯了,学习学习学习

class Solution {
    List<List<Integer>> res;
    LinkedList<Integer> arr = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        res = new ArrayList<>();
        backtracking(n, k, 1);
        return res;
    }
    public void backtracking(int n, int k,int s){
        if(arr.size() == k){
            res.add(new ArrayList(arr));
            return;
        }
        //剪枝,只需要遍历到还需要多少个元素即可,比如k=3,arr.size()=0,遍历到n-2,因为要给下层留2个数来取。
        for(int i = s; i <= n - (k - arr.size()) + 1; i++){
            //每一层取一个数,然后去下一层
            arr.addLast(i);
            backtracking(n, k, i+1);
            //回溯
            arr.removeLast();
        }
    }
}

上一篇:回溯算法精讲


下一篇:LeetCode 77. 组合