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

思路：

先排序。

第一种思路：从左往右遍历，将第一个数字和第二个数字固定下来，然后对后面的数字进行二分查找，这样子的时间复杂度是o（n * n * log(n)）；

第二种思路：从左往右遍历，只固定第一个数字，然后同时向右和向左移动指针，判断三个数字相加的和是否为0。细节处理在于如果第一个数字重复，则无须再考虑，同时找到一组解之后要忽略掉重复的，注意避免陷入死循环。时间复杂度是o（n * n）。

 

代码：

class Solution {
public:
    vector<vector<int> > threeSum(vector<int>& nums) {
        vector<vector<int> > vec;
        sort(nums.begin(),nums.end());
        for(int i = 0;i < ((int)nums.size() - 2);i++)
        {
            if(i > 0 && nums[i] == nums[i - 1])
                continue;
            int temp = nums[i];
            for(int j = i + 1,k = (int)nums.size() - 1;j < k;)
            {
                if(temp + nums[j] + nums[k] == 0)
                {
                    int arr[3] = {temp,nums[j],nums[k]};
                    vector<int> ans(arr,arr+3);
                    vec.push_back(ans);
                    while(j < (int)nums.size() - 1 && nums[j] == nums[j + 1]) j++;
                    while(k > j && nums[k] == nums[k - 1]) k--;
                    j++;
                    k--;
                }
                else if(temp + nums[j] + nums[k] > 0) k--;
                else j++;
            }
        }
        return vec;
    }
};