【leetcode】15.三数之和（python+转为两数之和+去重）

zzamx

于 2023-07-08 09:42:16 发布

阅读量384

点赞数

分类专栏： Leetcode Python 文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/xiaoyue_/article/details/131608279

版权

Leetcode 同时被 2 个专栏收录

119 篇文章 2 订阅

订阅专栏

Python

70 篇文章 0 订阅

订阅专栏

在这里插入图片描述

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        # 思路：转为两数之和
        # for循环遍历先固定一个数字a，寻找另外两个数字之和=-a（双指针）
        # 难点：去重{1. 排序； 2.加规则约束}
        nums = sorted(nums)
        res = []
        for i in range(len(nums)):
            if nums[i] > 0: # 剪枝，nums[i]若为正数，后边的也都是正数，没有匹配的了。
                return res
            if i > 0 and nums[i] == nums[i-1]: # 写 nums[i] == nums[i+1]，会越界
                continue
            tar = 0 - nums[i]
            left, right = i + 1, len(nums) - 1
            while left < right:
                summ = nums[left] + nums[right]
                if summ == tar:
                    res.append([nums[i], nums[left], nums[right]])
                    while left < right and nums[left] == nums[left+1]:
                        left += 1
                    while left < right and nums[right] == nums[right-1]:
                        right -= 1
                    left += 1
                    right -= 1
                elif summ < tar:
                    left += 1
                else:
                    right -= 1
        return res