15. 3Sum

题目：

给定n个整数的数组nums，nums中是否有元素a，b，c，a + b + c = 0？ 找到数组中所有唯一的三元组，它们的总和为零。（解决方案不得包含重复的三元组）

示例：

Given array nums = [-1, 0, 1, 2, -1, -4],

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

思路：

要找出三个数且和为0，那么除了三个数全是0的情况之外，肯定会有负数和正数，我们还是要先fix一个数，然后去找另外两个数，我们只要找到两个数且和为第一个fix数的相反数就行了，如果能更有效的定位呢？我们肯定不希望遍历所有两个数的组合吧，所以如果数组是有序的，那么我们就可以用双指针以线性时间复杂度来遍历所有满足题意的两个数组合。我们对原数组进行排序，然后开始遍历排序后的数组，这里注意不是遍历到最后一个停止，而是到倒数第三个就可以了。这里我们可以先做个剪枝优化，就是当遍历到正数的时候就break，为啥呢，因为我们的数组现在是有序的了，如果第一个要fix的数就是正数了，那么后面的数字就都是正数，就永远不会出现和为0的情况了。然后我们还要加上重复就跳过的处理，处理方法是从第二个数开始，如果和前面的数字相等，就跳过，因为我们不想把相同的数字fix两次。对于遍历到的数，用0减去这个fix的数得到一个target，然后只需要再之后找到两个数之和等于target即可。我们用两个指针分别指向fix数字之后开始的数组首尾两个数，如果两个数和正好为target，则将这两个数和fix的数一起存入结果中。然后就是跳过重复数字的步骤了，两个指针都需要检测重复数字。如果两数之和小于target，则我们将左边那个指针i右移一位，使得指向的数字增大一些。同理，如果两数之和大于target，则我们将右边那个指针j左移一位，使得指向的数字减小一些，C++代码如下：

class Solution {
public:
    vector<vector<int> > threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int> > res;

        if(nums.size() < 3)
            return res;
        for(int i = 0;i<nums.size();i++)
        {
            if(nums[i] > 0)
                break;
            if(i >= 1 && nums[i] == nums[i-1])
                continue;
            int tar = 0-nums[i];
            int p = i+1,q = nums.size()-1;
            while(p < q)
            {
                if(nums[p] + nums[q] == tar)
                {
                    vector<int> v;
                    v.push_back(nums[i]);v.push_back(nums[p]);v.push_back(nums[q]);
                    res.push_back(v);
                    while(p<q && nums[p] == nums[p+1])++p;
                    while(p<q && nums[q] == nums[q-1])--q;
                    p++;q--;
                }

                if(nums[p] + nums[q] < tar)p++;
                if(nums[p] + nums[q] > tar)q--;
            }
        }
        return res;
    }
};