35.leetcode15_3Sum

1.题目描述

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.

如果S中有a,b,c满足a+b+c==0，返回所有的满足这个条件的[a,b,c]，而且不能重复

2.题目分析

如果采用暴力搜索法，三指针从头至尾依次遍历所有元素，那么复杂度为O(n3)；如果先将S排序，先固定一个元素，然后再双指针分别从固定元素之后的首元素和末元素开始遍历，时间复杂度为O(n2)

3.解题思路

找出所有的[a,b,c]组合不难，难点就在如何避免重复的[a,b,c]。一开始采用的是not in进行搜索，然后超时了。

①失败的代码：

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result=[]
        temp=[]
        nums.sort()
        i=0
        l=len(nums)
        while i<l-2:
            left=i+1
            right=l-1
            if nums[i]>0 or nums[right]<0:
                break
            while left<right and nums[left]+nums[i]<=0 and nums[right]>=0:
                s=nums[i]+nums[left]+nums[right]
                if s==0 and [nums[i],nums[left],nums[right]] not in result: #问题就在这里，not in 使搜索的复杂度随着result元素的增加而增加
                    temp=[nums[i],nums[left],nums[right]]
                    result.append(temp)
                    left+=1
                    right-=1
                elif s>0:
                    right-=1
                else:
                    left+=1
            i+=1
        return result

昨天晚上找了一个小时没有发现这个bug，ε=(´ο｀*)))唉。今天早晨明白这个bug以后，重新改了一下。

②成功的代码

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result=[]
        nums.sort()
        l=len(nums)
        for i in range(0,l-2):
            if i>0 and nums[i]==nums[i-1]: #遇见重复的元素，直接跳过就好了
                continue
            left=i+1
            right=l-1
            if nums[i]>0 or nums[right]<0:
                break
            while left<right and nums[left]+nums[i]<=0: #不需要not in搜索result数组元素
                s=nums[i]+nums[left]+nums[right]
                if s==0:
                    result.append([nums[i],nums[left],nums[right]])
                    left+=1
                    right-=1
                    while left<right and nums[left]==nums[left-1]:left+=1 #跳过
                    while left<right and nums[right]==nums[right+1]:right-=1 #跳过
                elif s>0:
                    right-=1
                else:
                    left+=1
        return result