[LeetCode]3Sum

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:

• Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
• 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)

利用Hash Table可将时间复杂度降为O(n^2)，原因是将第三个循环遍历改成了Hash Table的判断key是否存在，和Two Sum类似。
首先进行排序，时间复杂度n*log(n)， 然后固定前两个值，查找是否存在符合条件的第三个值。
为了节约时间，已经出现的第一个数字如果再次出现可以直接continue了。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        vector<int> temp;
        sort(nums.begin(),nums.end());
        unordered_map<int,int> showed;
        for(int i=0;i<nums.size();i++)
        {
            showed[nums[i]] = i;
        }
        for(int i=0;i<nums.size();i++)
        {
            if((i>0)&&(nums[i]==nums[i-1])) continue;
            for(int j=i+1;j<nums.size();j++)
            {
                if((j>(i+1))&&(nums[j]==nums[j-1])) continue;
                int num1=nums[i];
                int num2=nums[j];
                int target = 0-num1-num2;
                if((showed.find(target)!=showed.end())&&(showed[target]>j))
                {
                    temp.push_back(num1);
                    temp.push_back(num2);
                    temp.push_back(target);
                    result.push_back(temp);
                    temp.clear();
                }
            }
        }
        return result;
    }
};

 也可以用二分查找的思路来完成。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        set<vector<int>> tempres;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++)
        {
            if((i>0)&&(nums[i]==nums[i-1])) continue;
            int begin = i+1;
            int end = nums.size()-1;
            int target = 0-nums[i];
            while(begin<end)
            {
                if((nums[begin]+nums[end])==target)
                {
                    vector<int> temp;
                    temp.push_back(nums[i]);
                    temp.push_back(nums[begin]);
                    temp.push_back(nums[end]);
                    tempres.insert(temp);
                    begin++;
                    end--;
                }
                else if((nums[begin]+nums[end])<target)
                {
                    begin++;
                }
                else
                {
                    end--;
                }
            }
        }
        set<vector<int>>::iterator it;
        for(it=tempres.begin();it!=tempres.end();it++)
        {
            result.push_back(*it);
        }
        return result;
    }
};