15、三数之和（python）

最新推荐文章于 2025-06-06 14:48:08 发布

原创最新推荐文章于 2025-06-06 14:48:08 发布 · 677 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

LeetCode 专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种高效算法，用于在整数数组中查找所有不重复的三元组，这些三元组的和等于零。通过排序和双指针技巧，避免了重复结果并显著提高了搜索速度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

代码（排序+双指针）：

class Solution:
    def threeSum(self, nums):
        if len(nums) < 3:                 # 若数组长度 < 3,直接返回空
            return []
        nums = sorted(nums)               # 排序
        re = []                           # 结果变量
        for i in range(len(nums)):                  # 遍历
            if nums[i] > 0:                         # 若 nums[i] > 0,则没有数字之和为0，直接返回re
                return re 
            if i > 0 and nums[i] == nums[i-1]:      # 排除重复数组
                continue
            L = i + 1                               # 左指针
            R = len(nums) - 1                       # 右指针
            while L < R:                                      # 遍历
                temp = nums[i] + nums[L] + nums[R]            # 三数之和
                if temp == 0:                                 # 若temp = 0
                    re.append([nums[i],nums[L],nums[R]])      # 添加到结果列表中
                    while L < R and nums[L] == nums[L+1]:     # 避免左指针下一个数字重复，跳过
                        L = L + 1
                    while L < R and nums[R] == nums[R-1]:     # 避免右指针下一个数字重复，跳过
                        R = R - 1
                    L = L + 1                       # 左指针右移，遍历
                    R = R - 1                       # 右指针左移，遍历
                elif temp > 0:                                # 若temp > 0
                    R = R - 1                                 # 则应该让三数之和小一些，故右指针左移
                else:                                         
                    L = L + 1                                 # 反之，应该让三数之和大一些，故左指针右移
        return re