视频讲解:https://www.bilibili.com/video/BV12V4y1V73A/?vd_source=a935eaede74a204ec74fd041b917810c
这道题的重点在于,输入存在重复元素,输出却不需要重复组合
一开始的计划是,排序后从当前位的下一位继续回溯遍历,但这样仍然存在重复元素,比如1,1,2,2,target为3,第一位结果有12,第二位也有12,这显然不对。
那么下一个想法就是,如果我的当前位与上一位重复了,那我就直接跳过这一位就好了,比如1,1,2,2,target为4,我在遍历第一个1时,会有112的组合,但是再回溯到第二个1时就会跳过了,这显然不是我们想要的。
因此就得出一个结论,需要去区分什么时候遍历到第二个重复的就不会跳过,这时候引入used数组,当数组其中一位置1表示用过了,置0表示未使用,如果当前位和前一位数值相等,但前一位的used数组为0,表示当前位是树层新遍历的,直接跳过,如果前一位used为1,表示为叶子层遍历的,可以保留。
cpp
class Solution {
private:
vector<vector<int >> result;
vector<int> path;
void backtracking(vector<int > &candidates, int target, int sum, int startIndex, vector<bool>& used)
{
//判断终止条件
if(sum > target)
{
return;
}
if(sum == target)
{
result.push_back(path);
return;
}
//单层遍历
for(int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++)
{
// used[i - 1] == true,说明同一树枝candidates[i - 1]使用过
// used[i - 1] == false,说明同一树层candidates[i - 1]使用过
// 要对同一树层使用过的元素进行跳过
if (i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) {
continue;
}
//计算sum
sum = sum + candidates[i];
//压入
path.push_back(candidates[i]);
used[i] = true;
//回溯遍历
backtracking(candidates, target, sum, i + 1, used);
//还原sum
sum = sum - candidates[i];
//出path
path.pop_back();
used[i] = false;
}
}
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
//初始化数组
result.clear();
path.clear();
vector<bool> used(candidates.size(), false);
sort(candidates.begin(), candidates.end());
/*
for(int i = 0; i<candidates.size();i++)
{
cout << candidates[i] << endl;
}*/
backtracking(candidates, target, 0, 0, used);
return result;
}
};