Leetcode 15.三数之和

最新推荐文章于 2024-08-29 16:23:51 发布

普通攻击往后拉

最新推荐文章于 2024-08-29 16:23:51 发布

阅读量238

点赞数 2

分类专栏： Python编程基础算法与数据结构文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_43483381/article/details/141601847

版权

Python编程同时被 2 个专栏收录

99 篇文章 0 订阅

订阅专栏

基础算法与数据结构

97 篇文章 5 订阅

订阅专栏

在这里插入图片描述

解法：两处去重+二分法

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        res = []
        visited_numi = [] # 第一处去重
        for i in range(len(nums) - 2):
            if nums[i] not in visited_numi:
                visited_numi.append(nums[i])
                sortnums = nums[i+1: ]
                l, r = 0, len(sortnums) - 1
                while r > l:
                    if sortnums[l] + sortnums[r] == - nums[i]: 
                        res.append([nums[i], sortnums[l], sortnums[r]])
                        while l < r and sortnums[l + 1] == sortnums[l]: l += 1 # 第二处去重
                        while l < r and sortnums[r - 1] == sortnums[r]: r -= 1
                        l += 1
                        r -= 1
                    elif sortnums[l] + sortnums[r] > - nums[i]: r -= 1
                    else: l += 1
        return res