leetcode查找表题目之15.三数之和

题目描述：

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

思路：排序+双指针

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>>res;
        //输入为空或者size大小小于3，返回为空
        //如果不考虑空，即不写为空的代码，后面代码,直接基于nums不为空的条件下，代码则会报错
        if(nums.size()<3)
            return res;
        //先排序
        sort(nums.begin(),nums.end());
        //i后面一定会有l和r,所以i<nums.size()-2
        for(int i=0;i<nums.size()-2;i++)
        {
            //左指针每次是从i+1的位置开始，可能会递增
            int l=i+1;
            //右指针从数组-1开始，可能会递减
            int r=nums.size()-1;
            //i后面一定会有l和r,nums已经排好序，所以只要nums[i]大于0，即可结束
            if(nums[i]>0)
                break;
            //排完序之后，为了不得到相同的元组，保持i（即元组第一个元素）不重复
            if(i>0&&nums[i]==nums[i-1])
                continue;
            while(l<r)
            {
                int tmp=nums[i]+nums[l]+nums[r];
                if(tmp==0)
                {
                    res.push_back({nums[i],nums[l],nums[r]});
                    //为了避免相同的元组，i不变，l和r必须改变
                    //因为给定一个值，i不变，改变一个l，r也需要改变
                    while(l<r&&nums[l]==nums[l+1]) l++;
                    while(l<r&&nums[r]==nums[r-1]) r--;
                    l++;
                    r--;
                } 
                else if(tmp>0)
                {
                    r--;
                }
                else l++;
            }
        }
        return res;
    }
};