leetcode15 三数之和

15 3sum

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

示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

思路
在2sum的基础上加一层循环

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() <= 2)
            return ans;
        sort(nums.begin(), nums.end());
        if(nums[0] > 0) 
            return ans;
        int key;
        for(int i1=0;i1<nums.size()-2;i1++)
        {
            if(i1!=0 && nums[i1] == nums[i1-1])
                continue;
            key = - nums[i1];
            int i2 = i1+1, i3 = nums.size() - 1;
            if(nums[i3-1] + nums[i3] < key)
                continue;
            while(i2<i3)
            {
                if(nums[i2] + nums[i3] == key)
                {
                    ans.push_back({nums[i1],nums[i2++],nums[i3--]});
                    while(i2<i3 && nums[i3] == nums[i3+1]) i3--;
                    while(i2<i3 && nums[i2] == nums[i2-1]) i2++;
                }
                else if(nums[i2] + nums[i3] < key)
                {
                    i2++;
                    while(i2<i3 && nums[i2] == nums[i2-1]) i2++;
                }
                else
                {
                    i3--;
                    while(i2<i3 && nums[i3] == nums[i3+1]) i3--;
                }
            }
        }
        return ans;
    }
};

里面循环判断可以简化
这是因为如果nums[i2]+nums[i3] == -nums[i1] 且 nums[i2]==nums[i2-1]，那么这个三元组一定已经放入了答案的数组中，所以循环调过就行。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() <= 2)
            return ans;
        sort(nums.begin(), nums.end());
        if(nums[0] > 0) 
            return ans;
        int key;
        for(int i1=0;i1<nums.size()-2;i1++)
        {
            if(i1!=0 && nums[i1] == nums[i1-1])
                continue;
            key = - nums[i1];
            int i2 = i1+1, i3 = nums.size() - 1;
            if(nums[i3-1] + nums[i3] < key)
                continue;
            while(i2<i3)
            {
                if(nums[i2] + nums[i3] < key)
                {
                    i2++;
                }
                else if(nums[i2] + nums[i3] > key)
                {
                    i3--;
                }
                else
                {
                    if(i2==i1+1 || nums[i2] != nums[i2-1])
                    {
                        ans.push_back({nums[i1],nums[i2],nums[i3]});
                    }
                    i2++;
                    i3--;
                }
            }
        }
        return ans;
    }
};