【代码随想录day 22】 力扣 40.组合总和II

视频讲解:https://www.bilibili.com/video/BV12V4y1V73A/?vd_source=a935eaede74a204ec74fd041b917810c

文档讲解:https://programmercarl.com/0040.组合总和II.html#思路

力扣题目: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,表示为叶子层遍历的,可以保留。

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;
    }
};
相关推荐
Wang153024 分钟前
Java排序
java
逸风尊者29 分钟前
XGBoost模型工程使用
java·后端·算法
一嘴一个橘子29 分钟前
MP 自定义业务方法 (二)
java
LUVK_37 分钟前
第七章查找
数据结构·c++·考研·算法·408
khalil102043 分钟前
代码随想录算法训练营Day-31贪心算法 | 56. 合并区间、738. 单调递增的数字、968. 监控二叉树
数据结构·c++·算法·leetcode·贪心算法·二叉树·递归
低客的黑调1 小时前
MyBatis-Plus-从 CRUD 到高级特性
java·servlet·tomcat
就像风一样抓不住1 小时前
Java 手机号校验工具类
java
lihihi1 小时前
P9936 [NFLSPC #6] 等差数列
算法
凤山老林1 小时前
26-Java this 关键字
java·开发语言
啊我不会诶1 小时前
2024ICPC西安邀请赛补题
c++·算法