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

Question

仅供参考的问题陈述：

给定一个包含 n 个整数的数组 S，S 中是否存在元素 a、b、c 使得 a + b + c = 0？查找数组中所有唯一的三元组，其总和为零。

注意：解决方案集不得包含重复的三元组。

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

一个解集是：[ [-1, 0, 1], [-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] 而是应该只将它添加到向量中一次。

Answer 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？