Leetcode: 15. 3Sum 三数之和

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.

Example:

given array S = [-1, 0, 1, 2, -1, -4],

A solution set is: [ [-1, 0, 1], [-1, -1, 2] ]

在给定数组S中找出所有的三元组，使得每个组中三个数之和为0。
本题传送门：https://leetcode.com/problems/3sum/

解决方案

先排序，从小到大选取第一个数，再在剩余区间内左右夹逼。另外，排除一些不可能的情况可以提速很多。
{ … , a , … , b→ , … ,←c , …}
时间复杂度：O(n^2)，空间复杂度：O(1)，运行时间：39ms，超过93.16% 的方案 !

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        if(nums.size()<3) return result;//不足3个元素，返回空
        sort(nums.begin(),nums.end());//排序
        const int target = 0;
        int a,b,c;//3个元素的下标
        for(a=0;a<nums.size()-2;a++){
            b=a+1;
            c=nums.size()-1;
            if(a>0 && nums[a]==nums[a-1])           continue;//跳过重复
            if(nums[a]+nums[a+1]+nums[a+2]>target)  break;//太大了
            if(nums[a]+nums[c]+nums[c-1]<target)    continue;//太小了
            while(b<c){
                int sum=nums[a]+nums[b]+nums[c];
                if(sum<target){
                    b++;
                    while(b<c && nums[b]==nums[b-1])    b++;//跳过重复
                }else if(sum>target){
                    c--;
                    while(b<c && nums[c]==nums[c+1])    c--;//跳过重复
                }else{
                    result.push_back({nums[a],nums[b],nums[c]});//得到一个解
                    b++;c--;
                    while(b<c && nums[c]==nums[c+1])    c--;//跳过重复
                    while(b<c && nums[b]==nums[b-1])    b++;//跳过重复
                }
            }
        }
        return result;
    }
};