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

思路

给定一个整型数组S，找到所有三个元素加起来为零的数。

思路1.

三个数加起来得是0，那么查找 0 - 前两个数之和 是否在数组S里，在就说明S里这三个数相加是0.

那么就插入二维数组中。

排重：新得到的数要和已经在二维数组里的数比较，不相同才可以插入到二维数组。发现超时。

思路2.

首先排序数组S，利用双指针一个在前，一个在后去比较加起来是否为零，然后双指针继续移动。

代码

class Solution {
public:
    vector< vector<int> > threeSum(vector<int>& nums) {
        int i, j, k;
        vector< vector<int> > res;

        if(nums.size() <= 2)
            return res; 

        sort(nums.begin(),nums.end());//排序从小到大

        for(i = 0; i < nums.size() - 2; i++)
        {
            j = i+1;
            k = nums.size() - 1;
            while(j < k)
            {
                vector<int> initial;
                if(nums[i] + nums[j] + nums[k] == 0)
                {
                    initial.push_back(nums[i]);
                    initial.push_back(nums[j]);
                    initial.push_back(nums[k]);
                    res.push_back(initial);
                    ++j;
                    --k;
                    while(j < k && nums[j-1] == nums[j])//向后移动了的值与之前相同
                    {
                        j++;
                    }
                    while(j < k && nums[k] == nums[k+1])
                    {
                        k--;
                    }
                }
                else if(nums[i] + nums[j] + nums[k] >= 0) //说明太大要缩小
                {
                    k--;
                }
                else
                    j++;
            }
            while(i < nums.size() - 1 && nums[i] == nums[i+1])
            {
                i++;
            }           
        }
        return res;
    }
};