这是一道经典的回溯算法题目,回溯的典型模板为一个for循环遍历,同时也作为回溯的出口,再加上一个递归来深度遍历。
具体到这道题的解法,用ans存储最后的答案,也就是二维数组,再用一个combine存储每次回溯得到的数组。对于回溯函数,以索引值idx来遍历candidates中的数,每加入一个candidates中的值,就判断combine中的值是否和等于目标值target,当递归到最后发现值并不等于target,及时退栈,再继续遍历其他的值,当idx等于candidates的大小,这是已经遍历完整个数组。
class Solution {
public:
void dfs(vector<int>& candidates,int target,vector<vector<int>>& ans,
vector<int>& combine,int idx){
if(idx == candidates.size()){
return;
}
if(target == 0){
ans.emplace_back(combine);
return;
}
dfs(candidates,target,ans,combine,idx + 1);
if(target - candidates[idx] >= 0){
combine.emplace_back(candidates[idx]);
dfs(candidates,target - candidates[idx],ans,combine,idx);
combine.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<vector<int>> ans;
vector<int> combine;
dfs(candidates,target,ans,combine,0);
return ans;
}
};
如果对回溯不怎么理解的可以参考n皇后问题作为基础。