题目描述
给定两个整数 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]]
思路
回溯 + 剪枝 + 栈(用双端队列ArrayDeque
实现)
- 回溯遍历所有可能的起点,并剪枝(限定下一个搜索的范围一定比当前数要大)
- 临时路径使用栈来存储,方便恢复现场(删除最后一个)
代码
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> res = new ArrayList<>();
// 双端队列 替代栈
Deque<Integer> path = new ArrayDeque<>();
backTrack(n, k, 1, res, path);
return res;
}
public void backTrack(int n, int k, int begin, List<List<Integer>> res, Deque<Integer> path) {
if (path.size() == k) {
//加入结果集
res.add(new ArrayList<>(path));
return;
}
//遍历所有起点 (开始的数)
for (int i = begin; i <= n; i ++ ) {
path.add(i);
//搜索下一个数,一定得比当前大(剪枝)
backTrack(n, k, i + 1, res, path);
//恢复现场
path.removeLast();
}
}
}