力扣 15. 三数之和

最新推荐文章于 2024-09-25 23:58:26 发布

团子大圆帅

最新推荐文章于 2024-09-25 23:58:26 发布

阅读量97

点赞数

文章标签：力扣数据结构与算法数组双指针

本文链接：https://blog.csdn.net/jpch89/article/details/115047171

版权

15. 三数之和

题目描述

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

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

示例 2：

输入：nums = []
输出：[]

示例 3：

输入：nums = [0]
输出：[]

提示：

0 <= nums.length <= 3000
-105 <= nums[i] <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

双指针

要点：

使用排序去重
使用两个 for 循环选择第一个数和第二个数，用一个 while 循环选择第三个数
注意：k = n - 1 如果放到 for j in range(i + 1, n - 1) 中，则只能通过 315 个测试用例。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        n = len(nums)
        res = []
        for i in range(n - 2):
            # 剪枝
            if nums[i] > 0:
                break
            # 保证第一个数不重复
            if i > 0 and nums[i] == nums[i - 1]:
                continue
            target = -nums[i]
            # 由于第二个数是递增的，所以 k 不用每次都从 n - 1 开始
            # 所以这一句不能放到下面的循环里，会影响性能
            k = n - 1
            for j in range(i + 1, n - 1):
                # 保证第二个数不重复
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                while j < k and nums[j] + nums[k] > target:
                    k -= 1
                if j == k:
                    break
                if nums[j] + nums[k] == target:
                    res.append([nums[i], nums[j], nums[k]])
        return res