题目链接
题目描述
给定两个整数 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
解题思路
回溯法
将组合问题抽象成如下树形结构,每次从集合中选取元素,可选择的范围随着选择的进行而收缩,调整可选择的范围。图中可以发现n相当于树的宽度,k相当于树的深度。图中每次搜索到了叶子节点,我们就找打了一个结果
回溯法三部曲
- 递归函数的参数和返回值
- n,k,startIndex,用来记录本层递归中,集合从哪里开始遍历
path
:用来存放符合条件的结果ans
:用来存放符合条件结果的集合
- 回溯函数终止条件
- 到达叶子节点。也就是
path
这个数组的大小达到k
,说明已经找打了一个子集大小为k
的组合了,在图中path
存的就是根节点到叶子节点的路径了。 - 此时用
ans
数组将结果保存起来,并终止本层递归
- 到达叶子节点。也就是
- 单层逻辑
- 用
for
循环进行横向遍历,具体来说是每次从startIndex
开始遍历,然后用path
保存取到的节点i
- 用
剪枝优化
如图所示,可以剪枝的地方就在递归中每一层的for循环选择的起始位置
- 如果for循环选择的起始位置之后的元素个数已经不足我们需要的元素个数了,那么就没有必要搜索了
AC代码
class Solution {
public static List<List<Integer>> combine(int n, int k) {
List<List<Integer>> ans = new ArrayList<>();
List<Integer> path = new ArrayList<>();
backTracing(n, k, 1, path, ans);
return ans;
}
private static void backTracing(int n, int k, int startIndex, List<Integer> path, List<List<Integer>> ans) {
//终止条件
if (path.size() == k) {
ans.add(new ArrayList<>(path));
return;
}
for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) {
path.add(i);
backTracing(n, k, i + 1, path, ans);
//回溯
path.remove(path.size() - 1);
}
}
}