给定一个数组 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]
]
思路:在90题的基础上进行剪枝,用一个sum来记录当前集合元素的和,当sum > target时,我们直接返回,这就是剪枝,当sum == target且当前item未在result中出现,就将其加入result
#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
class Solution{
public:
vector<vector<int> > combinationSum2(vector<int> &nums,int target) {
vector<vector<int> > result;
set<vector<int> > res_set;
vector<int> item;
sort(nums.begin(),nums.end());
generate(0,target,nums,item,res_set,result);
return result;
}
void generate(int i,int target,vector<int> &nums,vector<int> &item,set<vector<int>> &res_set,
vector<vector<int> > &result) {
if(i == nums.size() || sum > target) {
return;
}
item.push_back(nums[i]);
sum+=nums[i];
if(sum == target && res_set.find(item) == res_set.end()) {
res_set.insert(item);
result.push_back(item);
}
generate(i+1,target,nums,item,res_set,result); //当前元素加入item,继续向后面所有元素递归
item.pop_back();
sum -= nums[i];
generate(i+1,target,nums,item,res_set,result); //当前元素不加入item,继续向后面所有元素递归
}
};
int main() {
vector<int> nums;
nums.push_back(10);
nums.push_back(1);
nums.push_back(2);
nums.push_back(7);
nums.push_back(6);
nums.push_back(1);
nums.push_back(5);
vector<vector<int> > result;
Solution s;
result = s.combinationSum2(nums,8);
for(int i = 0; i < result.size(); i++) {
for(int j = 0; j < result[i].size(); j++) {
cout<<"["<<result[i][j]<<"]";
}
cout<<endl;
}
return 0;
}