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

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

回溯法的代码模板

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

    for (选择：本层集合中元素（树中节点孩子的数量就是集合的大小）) {
        处理节点;
        backtracking(路径，选择列表); // 递归
        回溯，撤销处理结果
    }
}
1
2
3
4
5
6
7
8
9
10
11
12

一、77. 组合

题目链接：https://leetcode.cn/problems/combinations/
思路：for循环横向控制可选择的空间，递归搜集数据直抵叶节点，此外还有剪枝要注意。

class Solution {
    List<List<Integer>> arrayLists = new ArrayList<>();
    List<Integer> list = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        backTracking(n, k, 1);
        return arrayLists;
    }
    void backTracking(int n, int k, int left) {
        if (list.size() == k) {
            arrayLists.add(new ArrayList<>(list));
            return;
        }
        for (int i = left; i <= n - (k - list.size()) + 1; i++) {
            list.add(i);
            backTracking(n, k, i+1);
            list.remove(list.size()-1);
        }
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19