Leetcode:组合
回溯法代码框架
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:
void backtrack(vector<vector<int>> &result,vector<int> &ans,int num,int n,int k){
if(ans.size()+(n-num+1)<k){
return;
}
if(ans.size()==k){
result.push_back(ans);
return;
}
ans.push_back(num);
//子结果的控制因素,如确定vector的起始元素是1、2,在后面变化元素防止重复
backtrack(result,ans,num+1,n,k);
ans.pop_back();
//在把末尾的元素弹出后,需要继续加入新元素,所以这里要加一个回溯
backtrack(result,ans,num+1,n,k);
}
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> result;
vector<int> ans;
backtrack(result,ans,1,n,k);
return result;
}
};

浙公网安备 33010602011771号