leetcode15. 三数之和

题目大意:求出集合中三个元素之和等于0,并且无重复的三元组

题目分析:本题直接用暴力搜索来枚举所有情况,时间复杂度为O(n^3),超时

下面使用第二种方法,时间复杂度降到O(n^2):它的思路是先将集合中的元素从小到大排序,然后确定三元组中的第一个元素为集合中的第一个元素,然后用指针j,k分别指向第二个元素和最后一个元素。若此时i+j+k=0,表示是符合条件的一种情况。若i+j+k>0,则表示现在和太大了,所以要将k前移。要是将j后移,结果不是会更大嘛。若i+j+k<0,则表示现在和太小了,将j后移。当i和j相遇时,这种情况已经遍历完了。然后将集合中第二个元素作为三元组中的第一个元素,j指向集合中第三个元素,k指向最后一个元素,类似的比较……,每增加一种情况之前,判断是否重复。

这样做了之后,发现还是超时的。原因出在判断重复这步上。我们可以通过一些步骤来化简。如果nums[i] = nums[i-1],那么之后得到的两个数和nums[i]组合在一块这种情况,和之后得到的两个数和nums[i-1]组合在一块这种情况 是重复的。还有在j后移,k前移时,若前后两个数(比如nums[j],nums[j+1])是相等的,在这种情况下也会产生重复。

代码展示:

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result = []
        nums.sort()
        for i in range(len(nums)-2):
            if i>0 and nums[i]==nums[i-1]:
                continue
            j = i+1
            k = len(nums)-1
            while j<k:
                temp = nums[i]+nums[j]+nums[k]
                if temp>0:
                    k = k-1
                elif temp<0:
                    j = j+1
                else:
                    result.append([nums[i],nums[j],nums[k]])
                    while j<k and nums[j]==nums[j+1]:
                        j = j+1
                    while j<k and nums[k]==nums[k-1]:
                        k = k-1     
                    j = j+1
                    k = k-1
        return result

猜你喜欢

转载自blog.csdn.net/jaster_wisdom/article/details/80012433