LeetCode练习——数组和字符串（三数之和）

最新推荐文章于 2024-01-03 08:01:14 发布

原创最新推荐文章于 2024-01-03 08:01:14 发布 · 473 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #python

LeetCode 专栏收录该内容

85 篇文章

订阅专栏

该博客介绍了如何通过排序和双指针算法解决LeetCode上的三数之和问题。代码实现了一个Python类方法，通过遍历排序后的数组，找到所有不重复的三元组使得它们的和为0。算法复杂度为O(n^2)，有效地避免了重复三元组的出现。

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。

官方解法：https://leetcode-cn.com/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/

解法一：排序 + 双指针（620ms/17.6MB）
在这里插入图片描述

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        nums.sort()
        ans = list()
        
        # 枚举 a
        for first in range(n):
            # 需要和上一次枚举的数不相同
            if first > 0 and nums[first] == nums[first - 1]:
                continue  # 跳过当次循环中剩下的语句，执行下一次循环。
            # c 对应的指针初始指向数组的最右端
            third = n - 1
            target = -nums[first]
            # 枚举 b
            for second in range(first + 1, n):
                # 需要和上一次枚举的数不相同
                if second > first + 1 and nums[second] == nums[second - 1]:
                    continue
                # 需要保证 b 的指针在 c 的指针的左侧
                while second < third and nums[second] + nums[third] > target:
                    third -= 1
                # 如果指针重合，随着 b 后续的增加
                # 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if second == third:
                    break  # 完全结束一个循环，跳出循环体执行循环后面的语句。
                if nums[second] + nums[third] == target:
                    ans.append([nums[first], nums[second], nums[third]])
        
        return ans