Leetcode #15: 3sum——避免重复

Leetcode #15: 3sum -- avoiding duplicate

本文关键字:Leetcode 3sum      更新时间:2023-10-16

问题说明:

给定一个包含n个整数的数组S, S中是否存在元素a, b, c使得a + b + c = 0?在数组中找到所有唯一的和为0的三元组

注意:解集不能包含重复的三元组。

例如,给定数组S = [-1, 0, 1, 2, -1, -4]

解集是:[[1,0,1],[1,2]]

下面是我的算法,工作合理,如预期的那样,但我不知道如何防止重复。我已经注释了我试图跟踪重复集合(三元组)的部分。

vector<vector<int>> threeSum(vector<int>& nums)
{
    vector< vector<int> > res;
    int a,b,c,start,end;
    int preva=0,prevb=0,prevc=0;  //variables to track elements of
                                  //triplet pushed into result 
                                  //vector the previous run.
    sort(nums.begin(),nums.end());
    int n=nums.size();
    for(int i=0; i<=n-3;i++){
        a = nums[i];
        start = i+1;
        end = n-1;
        while (start < end){
            b = nums[start];
            c = nums[end];
            if (a+b+c == 0)
            {
                if ((a!=preva)&&(b!=prevb)&&(c!=prevc)) //check if duplicate
                    res.push_back({a,b,c}); //add triplet 
                                            //to res vector if not
                                            //present.
                end = end - 1;
                preva=a;
                prevb=b;
                prevc=c;
               }
            else if (a+b+c > 0)
                end = end - 1;
            else
                start = start + 1;
        }
    }
    return res;
}

,

你的答案:[[-1,-1,2]]

不匹配

期望答案:[[-1,-1,2],[-1,0,1]]

我错过了添加[-1,0,1]完全当它应该被添加到向量只有一次

考虑以下if语句中的条件表达式:

            if ((a!=preva)&&(b!=prevb)&&(c!=prevc)) //check if duplicate

这将只在a,b,c不匹配preva,prevb,prevc推入结果;在[-1,0,1]的情况下,我们最终得到a = -1[-1,-1,2]匹配的preva = -1。此外,这只检查前面的解决方案。

相反,您应该确定一种顺序无关的方法来存储这些结果,并让容器本身处理重复的结果——也许是std::set而不是vector ?