【LeetCode刷题】15 三数之和（做的很坎坷，本质是双指针）

D2O

已于 2022-08-05 15:22:09 修改

阅读量83

点赞数

文章标签： leetcode 算法数据结构

于 2022-08-04 10:21:30 首次发布

本文链接：https://blog.csdn.net/D2Ooo/article/details/126154463

版权

三数之和排序双指针算法优化去重

关键词由CSDN通过智能技术生成

文章目录

版本1：没有充分利用排序信息，运行超时
版本2：用了双指针的方法，但是利用函数去重
版本3：双指针，进循环判断是否重复

版本1：没有充分利用排序信息，运行超时

from typing import List


def threeSum(nums: List[int]) -> List[List[int]]:
    res = []
    nums.sort()
    for r in range(len(nums)):
        if r!=len(nums)-1:
            if nums[r]==nums[r+1]:
                continue
        l = 0
        while l + 1 < r:
            if l!=0:
                if nums[l]==nums[l-1]:
                    l += 1
                    continue
            another = 0 - nums[l] - nums[r]
            if another in nums[l + 1: r]:
                res.append([nums[l], another, nums[r]])
            l += 1
    return res


if __name__ == '__main__':
    nums = [-1,0,1,2,-1,-4,-2,-3,3,0,4]
    ans = threeSum(nums)
    print(ans)

在这里插入图片描述

版本2：用了双指针的方法，但是利用函数去重

class Solution:
   def threeSum(self, nums: List[int]) -> List[List[int]]:
       n = len(nums)
       nums.sort()
       res = []
       for a in range(n - 2):
           b, c = a + 1, n - 1
           while b < c:
               temp =nums[a]+ nums[b] + nums[c]
               if temp == 0:
                   candidate=[nums[a], nums[b], nums[c]]
                   if candidate not in res:
                       res.append(candidate)
                   # res.append([nums[a], nums[b], nums[c]])
                   b += 1
                   c -= 1
               elif temp < 0:
                   b += 1
               else:
                   c -= 1
       return res

在这里插入图片描述

版本3：双指针，进循环判断是否重复

class Solution:
   def threeSum(self, nums: List[int]) -> List[List[int]]:
       n = len(nums)
       nums.sort()
       res = []
       for a in range(n - 2):
           # 对a进行去重操作
           if a != 0:
               if nums[a] == nums[a - 1]:
                   continue
           b, c = a + 1, n - 1

           while b < c:
               # 对b进行去重操作
               if b != a + 1:
                   if nums[b] == nums[b - 1]:
                       b += 1
                       continue
               # 对c进行去重操作
               if c != n - 1:
                   if nums[c] == nums[c + 1]:
                       c -= 1
                       continue
               temp =nums[a]+ nums[b] + nums[c]
               if temp == 0:
                   res.append([nums[a], nums[b], nums[c]])
                   b += 1
                   c -= 1
               elif temp < 0:
                   b += 1
               else:
                   c -= 1
       return res