leetcode-15. 3Sum

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路：

给一个数组，求出三个数和为0的组合，结果不能包含重复组合，比如例子里返回两个[-1,0,1]是不行的。这个题目可以转换为leetcode第一题two sum来做(不完全一样，不需要保存索引).

two sum:数组里两数之和等于第三个指定数target，返回那两个加数，可以现从小到大排序，在用两个指针，一个left指向头，一个right指向尾，如果nums[left]+nums[right]==target,就找到了，如果<则left+1，如果>则right-1。

three sum：先从小到大排序，然后用三个指针，比如这里求a+b+c=0可以转化a+b=-c，第一个指针指向c，然后在其余元素里用two sum的方法找组合，因为要去重，所以排序以后先拿第一个数当c，然后用后面数组元素做two sum，然后用第二个数当c，在用后面数组元素做two sum，即数组ijklmn里先i当c，从jklmn里求two sum，在j当c，从klmn里two sum。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int length=nums.size();
        vector<vector<int>>ans_list;
        for(int i=0;i<length-2;++i){
            if(i>=1&&nums[i]==nums[i-1]){//防止重复的-1 0 1这种情况
                continue;
            }
            int left=i+1,right=length-1;
            while(left<right){
                if(nums[left]+nums[right]==-nums[i]){
                    vector<int>ans;
                    ans.push_back(nums[i]);
                    ans.push_back(nums[left]);
                    ans.push_back(nums[right]);
                    ans_list.push_back(ans);
                    while(left<right && nums[left]==nums[left+1]) //防止重复
                        left++;  
                    while(left<right && nums[right]==nums[right-1]) 
                        right--;
                    left++;
                    right--;
                }else if(nums[left]+nums[right]<-nums[i]){
                     left++;
                }else{
                    right--;
                }
            }
        }
        return ans_list;
    }
};