LeetCode每日一题 40.组和总和II 回溯dfs+去重
题目描述
给定一个数组 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]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-ii
解题思路
今天这道题和做题的思路差不多,都是用回溯法dfs加上去重。本题中需要注意的一点是输入的candidates 序列中有重复的数字,这也就会造成重复的解答。如果继续使用昨天的方法(可参见上一篇博客,链接),就会造成重复,因为昨天的题中给出的candidates中的数组没有重复,为避免重复昨天的题中是跳过已经完全·搜索过的数字,但如果candidates中有重复的数字,这个方法就依旧会产生重复。如 [2,2,2,3],target=5,搜过第一个2后,即使跳过第一个2(第一次搜索的结果应该是[2,3]),从第二个2开始搜索还是会出现[2,3]这个答案。
解决这个重复问题,第一种方法是使用stl种的set因为set集合中的元素会自动去重。当时这样的效率会比较低。可以思考产生重复的根本原因:有相等的数字,所以可以跳过相等的数字来解决重复的问题。注意出现连续相等的数字时,搜索第一个时就会将所有的情况枚举一边,比如上边举的例子,第一次就会搜索一个2,两个2,三个2的情况,所以出现连续相等的数字时可以跳过除第一个意外的数来避免重复,这就是第二种方法:
本题还需注意将输入的candidates排个序,因为输入的candidates是乱序的,相等的数不一定连续。
题解代码
class Solution {
public:
void dfs(vector<vector<int>> &ans,vector<int> &temp,vector<int> &candidates,int target,int sum,int startIndex){
if(sum==target){
ans.push_back(temp);
return;
}
if(sum>target){
return;
}
for(int i=startIndex;i<candidates.size();i++){
if(i>startIndex&&candidates[i]==candidates[i-1]) continue;//跳过连续相等的数
temp.push_back(candidates[i]);
sum+=candidates[i];
dfs(ans,temp,candidates,target,sum,i+1);
temp.pop_back();
sum-=candidates[i];
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<vector<int>> ans;
vector<int> temp;
sort(candidates.begin(),candidates.end());
dfs(ans,temp,candidates,target,0,0);
return ans;
}
};