77. 组合
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
提示:
1 <= n <= 20
1 <= k <= n
思路
回溯算法模板框架如下:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
剪枝
解法
class Solution {
List<List<Integer>> result = new ArrayList<>();
List<Integer> list = new ArrayList<>();
public List<List<Integer>> combine(int n, int k) {
if (k == 0) {
return result;
}
backtracking(1, n, k);
return result;
}
public void backtracking(int start, int n, int k) {
if (k == 0) {
result.add(new ArrayList(list));
return;
}
for (int i = start; i <= n; i++) {
// 处理节点
list.add(i);
backtracking(i + 1, n, k - 1);
// 回溯,撤销处理的节点
list.remove(list.size() - 1);
}
}
}