【LeetCode】三数之和【排序，固定一个数，然后二分寻找】

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
   [-1, 0, 1],
   [-1, -1, 2]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum

分析：

暴力法的复杂度是O(N^3),肯定会超时

先将数组排序，假设k1<=k2<=k3,先固定k1,然后通过双指针法在k1的后面寻找k2和k3

可以利用的性质：

1.如果k1大于0，则三数之和肯定无法等于0，可以跳过该k1

2.如果v[i]==v[i-1]，那么i-1可以跳过，因为这样必然会导致重复结果

可以采用set去除重复结果

时间复杂度：O(N^2)

排序的复杂度是O(N*log N),但是后续固定看+双指针的复杂度是O(N^2),二者取大的，所以时间复杂度是O(N^2)

空间复杂度：快排需要，最好情况：O(log N),最坏情况：O(N)

 

code：

class Solution {
public:
vector<vector<int> > threeSum(vector<int>& a)
{
    vector<vector<int> > vv;
    vector<int> v;
    set<vector<int> > s;
    set<vector<int> >::iterator it;
    int n=a.size();
    if(n<3)
        return vv;
    sort(a.begin(),a.end());
    for(int k=0;k<n-2;k++)
    {
        if(a[k]>0)
            continue;
        if(k>0&&a[k-1]==a[k])
            continue;
        int l=k+1;
        int h=n-1;
        //cout<<"k="<<k<<"l="<<l<<" h="<<h<<endl;
        s.clear();
        while(l<h)
        {
            int ans=a[k]+a[l]+a[h];
            //cout<<"l="<<l<<" h="<<h<<"ans="<<ans<<endl;
            if(ans==0)
            {
                v.clear();
                v.push_back(a[k]);
                v.push_back(a[l]);
                v.push_back(a[h]);
                s.insert(v);
                l++;
                h--;
            }else if(ans<0)
            {
                l++;
            }else if(ans>0)
            {
                h--;
            }
        }
        if(s.size()!=0)
        {
            for(it=s.begin();it!=s.end();it++)
            {
                vv.push_back(*it);
            }
        }
    }
    return vv;
}
};

 

golang实现

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40

func threeSum(nums []int) [][]int {     if len(nums)<3{         return [][]int{}     }     var result [][]int       // 排序 k1<=k2<=k3     sort.Ints(nums)       // 固定k1，双指针二分寻找k2和k3     for i:=0;i<len(nums)-2;i++{         if nums[i]>0{             continue         }         if i>0&&nums[i]==nums[i-1]{             continue         }                   l,r:=i+1,len(nums)-1         for l<r{             value:=nums[i]+nums[l]+nums[r]             if value==0{                 result=append(result,[]int{nums[i],nums[l],nums[r]})                 for l<r && nums[l]==nums[l+1]{                     l++                 }                 for l<r && nums[r]==nums[r-1]{                     r--                 }                 l++                 r--             }else if value<0{                 l++             }else if value>0 {                 r--             }         }     }     return result }

 

　　

 

TRANSLATE with x

English

Arabic	Hebrew	Polish
Bulgarian	Hindi	Portuguese
Catalan	Hmong Daw	Romanian
Chinese Simplified	Hungarian	Russian
Chinese Traditional	Indonesian	Slovak
Czech	Italian	Slovenian
Danish	Japanese	Spanish
Dutch	Klingon	Swedish
English	Korean	Thai
Estonian	Latvian	Turkish
Finnish	Lithuanian	Ukrainian
French	Malay	Urdu
German	Maltese	Vietnamese
Greek	Norwegian	Welsh
Haitian Creole	Persian

 

TRANSLATE with

COPY THE URL BELOW

Back

EMBED THE SNIPPET BELOW IN YOUR SITE

Enable collaborative features and customize widget: Bing Webmaster Portal

Back