【代码随想录day 22】 力扣 40.组合总和II
视频讲解:https://www.bilibili.com/video/BV12V4y1V73A/?vd_source=a935eaede74a204ec74fd041b917810c
文档讲解:https://programmercarl.com/0040.%E7%BB%84%E5%90%88%E6%80%BB%E5%92%8CII.html#%E6%80%9D%E8%B7%AF
力扣题目:https://leetcode.cn/problems/combination-sum-ii/
这道题的重点在于,输入存在重复元素,输出却不需要重复组合
一开始的计划是,排序后从当前位的下一位继续回溯遍历,但这样仍然存在重复元素,比如1,1,2,2,target为3,第一位结果有12,第二位也有12,这显然不对。
那么下一个想法就是,如果我的当前位与上一位重复了,那我就直接跳过这一位就好了,比如1,1,2,2,target为4,我在遍历第一个1时,会有112的组合,但是再回溯到第二个1时就会跳过了,这显然不是我们想要的。
因此就得出一个结论,需要去区分什么时候遍历到第二个重复的就不会跳过,这时候引入used数组,当数组其中一位置1表示用过了,置0表示未使用,如果当前位和前一位数值相等,但前一位的used数组为0,表示当前位是树层新遍历的,直接跳过,如果前一位used为1,表示为叶子层遍历的,可以保留。
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;}//计算sumsum = sum + candidates[i];//压入path.push_back(candidates[i]);used[i] = true;//回溯遍历backtracking(candidates, target, sum, i + 1, used);//还原sumsum = sum - candidates[i];//出pathpath.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;}
};