三数之和问题（leetcode15）

三数之和问题（leetcode15）

找出一个数组中是否存在三个元素a,b,c，使得a,b,c=0。找出满足条件切 a，b，c不重复的三元组。找到的三元组也不能重复。

1. 数组长度<3,不存在这样的三元组。
2. 对数组进行排序，遍历数组，当前元素位置为 i，设定左右指针，左指针 $l$ 指向第一个比当前元素的大的位置，右指针 $r$ 指向数组末尾的一个位置。
   • 如果 nums[i]+nums[l]+nums[r]==0, 这个时候如果要继续走下去，需要消除重复的情况。如果当前 nums[l]=nums[l+1] 或者 nums[r]=nums[r-1]， 则需要继续移动指针至 第一个不和当前指针元素相同的元素。
   • 如果 nums[i]+nums[l]+nums[r]<0, 左指针 $l$ 右移。
   • 如果 nums[i]+nums[l]+nums[r]>0,右指针 $r$ 左移。

解法：排序+双指针

vector<vector<int>> threeSum(vector<int>& nums) {
    int n=nums.size();
    vector<vector<int>> res={};
    if(nums.empty() || n<3)
        return {};
    sort(nums.begin(),nums.end());
    for(int i=0;i<n-2;i++)
    {
        if(nums[i]>0)
            return res;
        if(i>0 && nums[i]==nums[i-1])
            continue;
        int l=i+1,r=n-1;
        while(l<r)
        {
            if(nums[i]+nums[l]+nums[r]==0)
            {
                res.push_back({nums[i],nums[l],nums[r]});
                while (l<r &&nums[l]==nums[l+1])
                    l++;
                while (l<r && nums[r]==nums[r-1])
                    r--;
                l+=1;
                r-=1;
            }
            else if(nums[i]+nums[l]+nums[r]>0)
            {
                r--;
            }else{
                l++;
            }
        }
    }
    return res;

}