题目描述
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
输入示例
n = 4, k = 2
输出示例
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
解题思路
我们使用回溯、深度优先遍历的思想,我们使用一个栈 path 来记录走过的路径,使用 begin 来记录当前来到的数字位置。递归过程如下:
解题代码
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> ans = new ArrayList<>();
if(k <= 0 || n < k) {
return ans;
}
// 从1开始是题目的设定
Deque<Integer> path = new ArrayDeque<>();
backtrack(ans, path, 1, n, k);
return ans;
}
public void backtrack(List<List<Integer>> ans, Deque<Integer> path, int begin, int n, int k) {
// 递归终止条件是path的长度等于k
if(path.size() == k) {
ans.add(new ArrayList<>(path));
return;
}
// 遍历可能的搜索起点
for(int i = begin; i <= n; i++) {
// 向路径变量里添加一个数
path.addLast(i);
// 下一轮搜索,设置起点要加1,因为组合数里不允许出现重复的元素
backtrack(ans, path, i+1, n, k);
// 深度优先遍历有回头过程,递归之后做逆向操作
path.removeLast();
}
}
}