15. 3Sum

static int wing=[]()
{
    std::ios::sync_with_stdio(false);
    cin.tie(NULL);
    return 0;
}();

class Solution 
{
public:
    vector<vector<int>> threeSum(vector<int>& nums) 
    {
        vector<vector<int>> res;
        int sz=nums.size();
        sort(nums.begin(),nums.end());
        int target,left,right,sum;
        for(int i=0;i<sz;i++)
        {
            if(nums[i]>0)
                break;
            if(i>0&&nums[i-1]==nums[i])
                continue;
            target=-nums[i];
            left=i+1;
            right=sz-1;
            while(left<right)
            {
                sum=nums[left]+nums[right];
                if(sum==target)
                {
                    res.push_back({nums[i],nums[left],nums[right]});
                    left++;
                    while(left<right&&nums[left]==nums[left-1])
                        left++;
                    right--;
                    while(left<right&&nums[right]==nums[right+1])
                        right--;
                }
                else if(sum<target)
                    left++;
                else
                    right--;
            }
        }
        return res;
    }
};

先排序，以一个元素为扫描基准，在该元素右侧查找两个元素，使得这两个元素的和为当前扫描元素的相反数。将三和问题转化为二和问题，时间复杂度O(n²)

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        int sz=nums.size();
        if(sz<3)
            return res;
        sort(nums.begin(), nums.end());
        int left,right,sum2,cursum2;
        int test=nums[sz-1]+nums[sz-2];
        int lim=sz-2;
        for(int i=0;i<lim;i++){
            if(nums[i]>0)
                return res;
            if(i>0&&nums[i]==nums[i-1])
                continue;
            if(nums[i]+test<0)
                continue;
            if(nums[i]+nums[i+1]+nums[i+2]>0)
                return res;
            sum2=-nums[i];
            left=i+1;
            right=sz-1;
            while(left<right){
                cursum2=nums[left]+nums[right];
                if(cursum2==sum2){
                    res.push_back({nums[i],nums[left],nums[right]});
                    ++left;
                    while(left<right&&nums[left]==nums[left-1])
                        ++left;
                    --right;
                    while(left<right&&nums[right]==nums[right+1])
                        --right;
                }
                else if(cursum2<sum2)
                    ++left;
                else
                    --right;
            }
        }
        return res;
    }
};

这个是带剪枝版本的，两段剪枝