39. 组合总和
难度中等
给你一个 无重复元素 的整数数组 candidates
和一个目标整数 target
,找出 candidates
中可以使数字和为目标数 target
的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates
中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target
的不同组合数少于 150
个。
破天荒地代码一次通过~,不过效率TCL
class Solution {
public:
vector<vector<int>> res;
void util(vector<int>& candidates,int i,int n,int sum,int target,vector<int> temp){
if(i==n || sum>target)
return;
if(sum==target){
res.push_back(temp);
return;
}
temp.push_back(candidates[i]);
util(candidates,i,n,sum+candidates[i],target,temp);
temp.pop_back();
util(candidates,i+1,n,sum,target,temp);
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
int n=candidates.size();
vector<int> temp;
util(candidates,0,n,0,target,temp);
return res;
}
};
用的回溯法,怎么耗时这么长哇,有简单的剪枝,emmm,再瞅瞅
永远相信官方解答下面评论区的大佬,参考其中一个评论,对代码进行优化,其实上一段代码之所以慢,是因为其对数字的选择和每个数字个数的试探都采用了dfs,而dfs实现方式是栈,数据多时,这个栈被撑得很大很大,不能及时回收,优化思路是,对数字种类的试探采用for循环,对数字个数的试探采用回溯,代码如下啦~
class Solution {
public:
vector<vector<int>> res;
void util(vector<int>& candidates,int begin,int n,int sum,int target,vector<int> temp){
if(sum>target)
return;
if(sum==target){
res.push_back(temp);
return;
}
for(int i=begin;i<n;i++){
if(sum+candidates[i]>target)
return;
temp.push_back(candidates[i]);
util(candidates,i,n,sum+candidates[i],target,temp);
temp.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
int n=candidates.size();
vector<int> temp;
util(candidates,0,n,0,target,temp);
return res;
}
};
看看结果
又刷了一题哎,又进步了哎,下一题见!😉👧