代码随想录 -- 哈希表 -- 三数之和

15. 三数之和 - 力扣(LeetCode)

思路:

        本题适合采用双指针法。

1. 先将数组 nums 升序排序

2. 初始状态:i = 0 ,指针 left = i+1 ,指针 right = len(nums)-1 (left 要 < right)

3. 如果 nums[i] > 0,则不存在满足条件的三元组;

    如果 nums[i] + nums[left] + nums[right] > 0,right 指针向前移动;

    如果 nums[i] + nums[left] + nums[right] < 0,left 指针向后移动;

    如果 nums[i] + nums[left] + nums[right] == 0,找到符合条件的三元组,加入res列表。

4. 去重:

    对于 i:当 i > 0 时,如果 nums[i] == nums[i-1],i += 1;

    对于 left:如果 nums[left] == nums[left-1],left += 1;

    对于 right:如果 nums[right] == nums[right+1],right -= 1.

class Solution(object):
    def threeSum(self, nums):
        nums.sort()
        res=[]
        for i in range(len(nums)):
            if nums[i]>0:
                return res
            if i>0 and nums[i]==nums[i-1]:
                continue
            left,right=i+1,len(nums)-1
            while left<right:
                sum = nums[i]+nums[left]+nums[right]
                if sum>0:
                    right-=1
                elif sum<0:
                    left+=1
                else:
                    res.append([nums[i],nums[left],nums[right]])
                    while left<right and nums[right]==nums[right-1]:
                        right-=1
                    while left<right and nums[left]==nums[left+1]:
                        left+=1
                    left+=1
                    right-=1
        return res

            

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值