给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
思路:这道题可以用回溯算法,回溯法的搜索过程就是一个树型结构的遍历过程,for循环用来横向遍历,递归的过程是纵向遍历。
- 确定回溯函数参数和返回值,回溯函数的参数一般是在写代码过程中确定的,不像递归函数,一开始就能确定有哪些参数,本题中,回溯函数的参数有n,k以及每一层for循环的起始搜索位置startIndex;回溯函数的返回类型一般都会void类型。
- 确定回溯函数的终止条件:如果搜索的长度达到了k,就把搜索路径加入结果集,然后return。
- 确定单层搜索的过程:for循环横向遍历,递归过程纵向遍历,for循环每次从startIndex开始遍历,然后用path保存每次取到的节点,递归函数不断调用自己往深处遍历,长度达到k,就要返回。然后递归函数下面要写回溯部分,撤掉本次处理的结果。
代码:
class Solution { //77. 组合
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(int n, int k, int startIndex) {
if (path.size() == k) {
result.push_back(path);
return;
}
for (int i = startIndex; i <= n - (k - path.size()) + 1; ++i) {
path.push_back(i);
backtracking(n, k, i + 1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return result;
}
};
参考资料:代码随想录