LeetCode算法题目：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]
]

---

分析：

此题目没有要求复杂度，可先对数据进行排序，然后再计算。

排序之后，我们就可以对数组用两个指针分别从前后两端向中间扫描了，如果是 2Sum，我们找到两个指针之和为target
就OK了，那 3Sum 类似，我们可以先固定一个数，然后找另外两个数之和为第一个数的相反数就可以了。

---

代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        const int target=0;
        auto last=nums.end();
        for(auto i = nums.begin();i<last-2;i++)
        {
            auto j=i+1;
            if(i>nums.begin()&&*i==*(i-1))  i++;
            auto k=last-1;
            while(j<k)
            {
                if(*i+*j+*k<target)
                {
                    j++;
                    while(*j==*(j-1)&&j<k)  j++;
                }
                else if(*i+*j+*k>target)
                {
                    k--;
                    while(*k==*(k+1)&&j<k) k--;
                }
                else
                {
                    result.push_back({*i,*j,*k});
                    j++;
                    k--;
                    while(*j==*(j-1)&&*k==*(k+1)&&j<k) j++;
                }
            }
        }
        return result;
    }
};