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]
]

本题要我们找到数组中所有3个数字相加为0的组合，而且不能有重复的。

解题思路：

因为给我们的数组是无序的，若是要找到所有组合就一定是要三层嵌套遍历的，所以要先进行排序。

排序完后的数组可以参照TowSum的解法：

　　将头一个数固定，在右端的数组中分别取最左和最右的数的下标为left和right。

　　当这两个数相加大于头一个数的负值时，右减，小于则反之。相当就加入二维数组，然后left加。

这道题比较头疼的是不能重复：

　　首先，如果头一个数重复的话可以直接跳过。

　　然后，若是后两个数重复的话可以和最后一次添加的三个数比较。因为这种解法各答案中的第一个数和第二个数都是升序，所以可以这么做。

代码：

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