原题地址(中等)
这个题就是很经典的回溯题。
我发现这类题有个共同的特点,(也可以说是回溯的特点吧),就是回溯函数中,一般都是如下结构:
- 减枝条件+终止条件;
- 递归条件:情况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 的 tempif (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;}};
