这个问题和“子集2”问题其实很像,比较暴力的方法是像做“子集2”一样,找出所有组合,看每个子集的元素之和是否等于target,但是这样肯定超时,复杂度O(2^n)
以示例1为例,target=8,而其中的元素10已经大于了8,所以,只要是包含10的子集肯定就是不满足条件的,再比如【2 7 6】某子集包含了这三个数的话,也是不满足条件的,所以遇到了这些不满足条件的,就不用再寻找下去了,直接return
程序主体和【子集2】差不多,只是在多加了几个判断,一个是return的条件,一个是子集之和是否等于target
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> res;
set<vector<int>> res_set;
vector<int> tmp;
sort(candidates.begin(),candidates.end());//别忘了这一步
com(0,candidates,tmp,res,res_set,0,target);
return res;
}
void com(int i ,vector<int>& candidates, vector<int>& tmp, vector<vector<int> >& res, set<vector<int> >& res_set,int sum,int target)
{
//如果当前tmp中的元素之和已经>target了,就没必要继续下去,(剪枝)
if(i>=candidates.size() || sum>target)
return;
tmp.push_back(candidates[i]);
sum +=candidates[i];
if(sum == target && res_set.find(tmp) == res_set.end())
{
res_set.insert(tmp);
res.push_back(tmp);
}
com(i+1,candidates,tmp,res,res_set,sum,target);
//递归没有nums[i]这个数的部分
sum -= candidates[i];
tmp.pop_back();
com(i+1,candidates,tmp,res,res_set,sum,target);
}
};