原题地址(中等)
这个题就是很经典的回溯题。
我发现这类题有个共同的特点,(也可以说是回溯的特点吧),就是回溯函数中,一般都是如下结构:
- 减枝条件+终止条件;
- 递归条件:情况1:考虑当前元素; 情况2:不考虑当前元素
也就是官方题解中说的[递归实现组合型枚举](https://leetcode-cn.com/problems/combinations/solution/zu-he-by-leetcode-solution/)
模板如下:
vector<int> temp;
void dfs(int cur, int n) {
if (cur == n + 1) {
// ...
return;
}
// 考虑选择当前位置
temp.push_back(cur);
dfs(cur + 1, n, k);
temp.pop_back();
// 考虑不选择当前位置
dfs(cur + 1, n, k);
}
代码里第一个减枝非常精髓,耗时可以提高非常多。
class Solution {
public:
vector<int> v;
vector<vector<int>> ans;
void dfs(int cur, int k, int n){
// 剪枝:temp 长度加上区间 [cur, n] 的长度小于 k,不可能构造出长度为 k 的 temp
if (v.size() + (n - cur + 1) < k)
return;
if(v.size() == k){
ans.push_back(v);
return;
}
v.push_back(cur);
dfs(cur+1, k, n);
v.pop_back();
dfs(cur+1, k, n);
}
vector<vector<int>> combine(int n, int k) {
dfs(1, k, n);
return ans;
}
};