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
  • -10^5 <= nums[i] <= 10^5

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

一解

使用排序 + 三指针。

  • first 指针代表第一个结果,second 和 third 分别代表第二,第三结果。
  • first 从数组的头开始遍历,thrid 从数组的尾部开始遍历,second 从 first 到 third 进行遍历。
class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        n = len(nums)
        res = []
        for first in range(n):
						# 跳过重复元素
            if first > 0 and nums[first] == nums[first - 1]:
                continue
            third = n - 1
            target = -nums[first]
            for second in range(first + 1, n):
								# 跳过重复元素
                if second > first + 1 and nums[second] == nums[second - 1]:
                    continue
                while second < third and nums[third] + nums[second] > target:
                    third -= 1
								# 没找到合适的数
                if second == third:
                    break
                if nums[third] + nums[second] == target:
                    res.append([nums[first], nums[second], nums[third]])
        return res