来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
题意:
给定两个整数 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(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
本题直接套用回溯算法的模板即可(图片转自代码随想录)
至于在for循环中的 i <= n - (k - path.size()) + 1
为什么这样写,可以看参考文章中剪枝优化部门的讲解。
本题Java代码:
class Solution {
List<List<Integer>> ans = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
private void backTracking(int n, int k, int start) {
if (path.size() == k) {
ans.add(new ArrayList<>(path));
return;
}
for (int i = start; i <= n - (k - path.size()) + 1; i++) {
path.add(i);
backTracking(n, k, i + 1);
path.removeLast();
}
}
public List<List<Integer>> combine(int n, int k) {
backTracking(n, k, 1);
return ans;
}
}