3sum 求三数之和等于0,不允许重复

https://leetcode.com/problems/3sum/

套路比较常见了,最重要的是去重。还是没法一次通过。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& a) {
        vector<vector<int>> ans;
        int n = a.size();
        if(n < 3)    return ans;
        
        sort(a.begin(),a.end());
        
        for(int i = 0; i < n - 2; i++)
        {
            int target = - a[i];
            
            for(int j = i + 1, k = n - 1; j < k; )
            {
                int sum2 = a[j] + a[k];
                
                if(sum2 == target)
                {
                    vector<int> tmp{a[i],a[j],a[k]};
                    ans.push_back(tmp);
                    
                    // 这一行做完以后,a[j]依然等于a[j-1]。
                    while(j+1 < n && a[j+1]==a[j]) j++;
                    j++;
                    while(k-1 >= 0&& a[k-1]==a[k]) k--;
                    k--;
                }
                else if(sum2 < target)
                {
                    j++;
                }
                else
                {
                    k--;
                }
            }
            //这一步并不总是执行的,只是走到所有的重复的最后。
            while(i+1 < n && a[i+1] == a[i]) i++;
        }
        
        return ans;
    }
};

去重的原理要好好想一下:
比如 -2 -2 -2 0 1 1 1 1 2
第一次i选中-2的时候,后面的组合有0 2,1 1.

这之后,如果再选第二个-2作为第一个数的话,后面依然会产生0 2 , 1 1,就发生了重复。
因为,如果第一次在集合A中找2,第二次相当于在A的子集中找2。第二次找出的结果肯定是第一次结果的子集。
所以,第一次选-2以后,第二次选应该找到第一个不是-2的数。这样去重。

 

转载于:https://www.cnblogs.com/buddho/p/10615341.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值