本题是一道使用回溯法解的题,在做的时候有了想法,但是就是写不对代码,所以写个博客记录一下,顺便加深一下理解。
一、题目
给定一个无重复元素的正整数数组 candidates 和一个正整数 target ,找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。
candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。
对于给定的输入,保证和为 target 的唯一组合数少于 150 个。
二、主要思路
使用回溯法,简单理解就是对若干个数,若其和小于target,则将其列为备选,说明还可能在放入一个数,使其和为最终的目标值target,此时将备选数组存入结果数组中,表示找到了一个合法解,然后清空备选数组,接着寻找下一个符合要求的解。
三、代码:
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.push_back(combine);
return;
}
dfs(candidates,target,ans,combine,idx+1);
if(target-candidates[idx]>=0){
combine.push_back(candidates[idx]);
dfs(candidates,target-candidates[idx],ans,combine,idx);
combine.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<int> combine;
vector<vector<int>> ans;
dfs(candidates,target,ans,combine,0);
return ans;
}
};
代码理解:
以 输入: candidates = [2,3,6,7], target = 7输出: [[7],[2,2,3]] 为例。
代码运行的主要步骤为:从第一个元素开始,不断执行 dfs(candidates,target,ans,combine,idx+1),直到dfs(candidates,target,ans,combine,4),此时idx==candidates.size(),返回至dfs(candidates,target,ans,combine,3),开始执行其后的语句,即判断target-candidates[3]是否大于等于0,若满足,则说明将candidates[3]加入备选数组,有可能得到合法解。此时,combine=[7],接着执行dfs(candidates,target-candidates[idx],ans,combine,idx)即dfs(candidates,0,ans,combine,3),由于新的target值为0,所以将combine存入ans中,返回至dfs(candidates,0,ans,combine,3),开始继续执行combine.pop_back()将combine清空,此时返回至dfs(candidates,target,ans,combine,2),开始接着向下执行,依次类推,最终即可得到全部符合要求的解。
combine数组的主要变化如下(未给出弹出的具体过程):
null ,null ,null ,null, null ,7 , 6 ,6 , 6, 3, 3, 3, 3, 3 3, 3 3, 3 3, 3 3, 2, 2, 2, 2, 2, 2 3, 2 3, 2 3 , 2 3, 2 2, 2 2, 2 2, 2 2, 2 2, 2 2 3, 2 2 2, 2 2 2, 2 2 2, 2 2 2, 2 2 2 。