题目大意是由一组数任意组合,使其和等于target,每个数只能用一次,而且答案不能重复。
题目在第39题上加了每个数字只能用一次的限制条件。需要对代码稍作修改。
这里用了一个isVisited数组来记录candidates中某个数是否在当前nums中
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> res;
vector<int> nums;
vector<bool> isVisited(candidates.size(), false);
sort(candidates.begin(), candidates.end());
backtrack(res, nums, candidates, target, 0, isVisited);
return res;
}
private:
void backtrack(vector<vector<int>> &res, vector<int> nums, const vector<int>& cans, int remain, int index, vector<bool>& isVisited) {
if (remain < 0) return;
else if (remain == 0) {
res.push_back(nums);
return;
}
else {
for (int i = index; i < cans.size(); ++i) {
if (i > 0 && cans[i] == cans[i-1] && !isVisited[i-1]) continue;//关键是这里的!isVisited[i-1]
isVisited[i] = true;
nums.push_back(cans[i]);
backtrack(res, nums, cans, remain-cans[i], i+1, isVisited);
nums.pop_back();
isVisited[i] = false;
}
}
}
};
//isVisited来表示candidates中某个数是否在当前nums中
//难以理解的是这里的!isVisited[i-1]
//只有当前数与前一个数相同,并且前一个数不在当前nums里面,说明的是:前一个数已经被考虑过了,而考虑当前这个数与考虑前一个数的情况是相同的,所以不应该再次考虑当前这个数,必须略过。
难以理解的是递归调用中的的!isVisited[i-1]:
只有当前数与前一个数相同,并且前一个数不在当前nums里面,说明的是:前一个数已经被考虑过了,而考虑当前这个数与考虑前一个数的情况是相同的,所以不应该再次考虑当前这个数,必须略过。
注意一个使用set的方法:
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> ret;
vector<int> temp;
sort(candidates.begin(), candidates.end());
helper(candidates, target, 0, 0, temp, ret);
return std::move(ret);
}
private:
void helper(vector<int>& nums, int target, int idx, int tempSum, vector<int>& temp, vector<vector<int>>& ret) {
if (tempSum > target)
return;
if (tempSum == target) {
ret.push_back(temp);
return;
}
if (idx == nums.size())
return;
unordered_set<int> si;
for (int i = idx; i < nums.size(); ++i) {
if (si.find(nums[i]) != si.end())
continue;
si.insert(nums[i]);
temp.push_back(nums[i]);
helper(nums, target, i+1, tempSum+nums[i], temp, ret);
temp.pop_back();
}
}
};
注意的是必须要排序,不然的话会发生元素相同但倒了一下的情况。
另外,既然排序了,不如用第一种used[]的方法。这里set只是一个备选项