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

分析

本题其实是找到两个和为特定值的组合的变形，即先确定第一个元素，然后问题就变成了在剩余元素中寻找两个和为第一个元素的负数的组合，由于题目要求找到所有的三元组，且三元组之间不重复，则代表第一个元素确定之后，与其值相同的后续元素都不能再作为第一元素出现，同样，当后两个元素的值确定之后，不能再出现相同的两个元素，所以先对整个数组排序，然后在寻找三元组的过程中去重。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());//先将数组排序
        int size=nums.size();
        if(size<3)
            return res;
        for(int i=0;i<size-2;++i){//选定第一个元素为nums[i]，下面就是从后续中找到两个元素之和为-nums[i]
            if(nums[i]>0)//当第一个元素大于0时，由于已经排序，不可能出现三元组之和等于0
                break;
            int target=-nums[i];
            int beg=i+1;//从i+1到size-1之间查找
            int end=size-1;
            while(beg<end){
                int sum=nums[beg]+nums[end];//查看sum与target之间的关系，小于则代表nums[beg]太小，大于则代表nums[end]过大，通过不断调整使得sum逼近target
                if(sum<target)
                    ++beg;
                else if(sum>target)
                    --end;
                else{//当sum=target时，构造三元组作为一个解，同时调整beg和end，忽略掉二者的重复元素，然后再查找下一个可能的解
                    res.push_back({nums[i],nums[beg],nums[end]});
                    while(beg<end&&nums[beg]==nums[beg+1])
                        ++beg;
                    while(beg<end&&nums[end]==nums[end-1])
                        --end;
                    ++beg;
                    --end;
                }
            }
            while(i+1<size&&nums[i+1]==nums[i])//以nums[i]为第一个元素的三元组查找完毕，下一步是忽略掉nums[i]的重复元素
                ++i;
        }
        return res;
    }
};

参考文章： Share my AC C++ solution, around 50ms, O(N*N), with explanation and comments