组合求和I
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入: candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
组合求和II
给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。 说明: 所有数字(包括目标数)都是正整数。 解集不能包含重复的组合。 示例
1:输入: candidates = [10,1,2,7,6,1,5], target = 8, 所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]
分析:
两题不同之处:一个是元素可以重复取用,一个是元素个数有限制
相同之处:不同路径下,元素排列顺序不同,但是元素相同算作一种情况,如,【1,3,2】和【1,2,3】,这时如果每层的搜索空间还是所有的数,会出现重复解,所以每一层的搜索空间要缩到从大于等于上一层数之后开始搜索。
class Solution {
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<vector<int>> res;
vector<int> path;
sort(candidates.begin(),candidates.end());
int sum =0,cur=0;//cur保证搜索空间是>=上一层添加的数,避免重复
dfs(candidates,res,path,target,sum,cur);
return res;
}
void dfs(vector<int>&nums,vector<vector<int>>& res,vector<int>&path,int target,int sum,int cur){
if(sum==target){
res.push_back(path);
return;
}
for(int i=cur;i<nums.size();i++){
if(sum+nums[i]>target)
break;
sum+=nums[i];
path.push_back(nums[i]);
dfs(nums,res,path,target,sum,i);
sum-=nums[i];
path.pop_back();
}
}
};
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
vector<vector<int>> res;
vector<int> path;
map<int,int>mp;
for(int i=0;i<candidates.size();i++){
mp[candidates[i]]++;
}
map<int,int>::iterator it;
vector<int>list;
for(it=mp.begin();it!=mp.end();it++){
list.push_back(it->first);
}
int sum=0,cur=0;
dfs(res,path,target,sum,mp,cur,list);
return res;
}
void dfs(vector<vector<int>> &res,vector<int> &path,int target,int sum,map<int,int>&mp,int cur,
vector<int>list){
if(sum>target){
//当前已经大于目标值就不必再往下搜索
return;
}
if(sum==target){
res.push_back(path);
return;
}
for(int i=cur;i<list.size();i++){
//为了避免重复,搜索空间是不重复的数
//不同排序算作一种情况时,每一层的搜索空间应该是从大于等于上一层的数往后搜
if(sum+list[i]>target)//如果当前的已经大于target,则后面搜索空间也不用再搜
break;
if(mp[list[i]]!=0){
sum+=list[i];
mp[list[i]]--;
path.push_back(list[i]);//要记得加入当前路径
dfs(res,path,target,sum,mp,i,list);
mp[list[i]]++;
sum-=list[i];
path.pop_back();
}
}
}
};