代码随想录算法训练营第二十四天| 77. 组合。

77. 组合

题目链接:组合

题目描述
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。

解题思路
本题是经典的回溯法解决的组合问题,回溯问题搞清楚纵向递归横向遍历即可,从题目可以看出横向是选取一个数,纵向是递归选取下一个数,如图所示。
因此按照卡哥的回溯模板

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

即可简单的写出此题的回溯算法。
代码随想录算法训练营第二十四天| 77. 组合。_第1张图片
此题的剪支可以从剩下的数字不足凑够k个数来进行剪支

代码实现
未剪支

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n+1,k,1);
        return res;
    }

    public void backtracking(int n, int k, int begin) {
        if (k == 0) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = begin; i < n; i++) {
            path.add(i);
            backtracking(n,--k,i+1);
            path.removeLast();
            ++k;
        }   
    }
}

剪支

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n+1,k,1);
        return res;
    }

    public void backtracking(int n, int k, int begin) {
        if (k == 0) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = begin; i < n-k+1; i++) {//修改此处i的范围为n-k+1
            path.add(i);
            backtracking(n,--k,i+1);
            path.removeLast();
            ++k;
        }   
    }
}

你可能感兴趣的:(算法)