三数之和python

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

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

示例 1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:

输入:nums = []
输出:[]
示例 3:

输入:nums = [0]
输出:[]

提示:

0 <= nums.length <= 3000
-105 <= nums[i] <= 105
错误代码:

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result=[]
        result1=[]
        if len(nums)<3:
            return []
        for i in range(len(nums)):
            for j in range(i+1,len(nums)):
                 for k in range(j+1,len(nums)):
                    if nums[i]+nums[j]+nums[k] == 0:
                        result.append([nums[i],nums[j],nums[k]])
        for i in result:
            i.sort()
        for i in result:
            if i not in result1:
                result1.append(i)
        return result1
                
                
                
nums=[-1,0,1,2,-1,-4]
s=Solution()
print(s.threeSum(nums))

时间花费太长了,有更优解。
正确解法:

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        result=[]
        for first in range(len(nums)):#遍历第一重循环
            if first >0 and nums[first]==nums[first-1]:#确保不与之前的重复
                continue
            third = len(nums)-1#从最右端开始
            target=-nums[first]
            for second in range(first+1,len(nums)):#遍历第二重循环,并注意是从第一个已经遍历了
                if second>first+1 and nums[second]==nums[second-1]:
                    continue#确保不重复
                while second<third and nums[third]+nums[second]>target:
                    third-=1#比0大故减小
                if second==third:
                    break#关键条件跳出
                if nums[third]+nums[second]==target:
                    result.append([nums[first],nums[second],nums[third]])
                                   
        return result

解析:首先进行排序,便于后续的第二重遍历和第三重遍历。采取双指针的方法,分别从左边与从右边向中间靠拢,直到出现第二个大于第三个跳出来即可。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值