LeetCode-15. 三数之和

15. 三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

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

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

双指针法

这道题关键点在于处理好去重，否则会超时。首先对nums数组由小到大排序，接着双指针法，由两边向中间移动，想想三个数的和为0，那必定是一个负两个正，或者两个负一个正，或者是一正一负一零。
去重:如果一个数与下一个数相同，那就向后或向前跳一位，比如:-1,-2,-2,0,2,和为0的三个数是-2，0，2，下标组合可能为1,3,4，也可能为2,3,4，所以当碰到-2时，下一位也是-2，那就可以跳过去。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.size()<3) return res;
        //set<vector<int>> s;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size()-2;i++)
        {
            if(i>0&&nums[i]==nums[i-1]) continue; //去重
            int l=i+1;
            int r=nums.size()-1;
            while(l<r)
            {
                int sum = nums[i]+nums[l]+nums[r];
                if(sum<0)
                {
                    l++;
                }
                else if(sum>0)
                {
                    r--;
                }
                else if(sum==0)
                {
                    while(l<r&&nums[l]==nums[l+1]) l++; //去重
                    while(l<r&&nums[r]==nums[r-1]) r--; //去重
                    vector<int> v={nums[i],nums[l],nums[r]};
                    //s.insert(v);
                    res.push_back(v);
                    l++;
                    r--;
                }
            }
        }
        //vector<vector<int>> vec(s.begin(),s.end());
        //res.assign(s.begin(),s.end());
        return res;
        
    }
};