39、组合总和
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
递归回溯什么的目前还不是很会,多找几个学习一下
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<vector<int>> ans;
vector<int> combine;
dfs(candidates, target, ans, combine, 0);
return ans;
}
};
40、组合总和 II
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
class Solution {
public:
void dfs(vector<int>& candidates, int target, vector<vector<int>>& ans, vector<int>& combine, int idx) {
if (target == 0) {
ans.push_back(combine);
return;
}
if (idx == candidates.size()) return;
// 无右分支,不可选同一个数字
//dfs(candidates, target, ans, combine, idx + 1);
// 左分支
for (int i=idx;i<candidates.size() && target - candidates[idx] >= 0;i++) {
if (i>idx && candidates[i]==candidates[i-1]) continue;//判重模板
combine.push_back(candidates[i]);
dfs(candidates, target - candidates[i], ans, combine, i+1);
combine.pop_back();//不是很理解,可能惯常操作
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> ans;
vector<int> combine;
sort(candidates.begin(),candidates.end());
dfs(candidates, target, ans, combine, 0);
return ans;
}
};
感觉用循环之后的理解难度比上一题的官方答案简单,就是其实都是对选定一个元素后,对其他元素都进行迭代即可,上一题还分为左右,没有必要,上一题的代码写成循环如下,感觉区别就是元素不能重复选择,则迭代进去的索引从下一个元素开始,同时需要判断答案的重复,就是之前一直用的判重模板:
class Solution {
public:
void dfs(vector<vector<int>>& res, vector<int>& path, vector<int>& candidates, int target, int sum, int start)
{
if (sum > target) return;
if (sum == target)
{
res.push_back(path);
return;
}
for (int i = start; i < candidates.size(); ++i)
{
path.push_back(candidates[i]);
dfs(res, path, candidates, target, sum + candidates[i], i);
path.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<vector<int>> res;
vector<int> path;
dfs(res, path, candidates, target, 0, 0);
return res;
}
};