15. 三数之和

题目描述

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

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

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

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

思路梳理

• 此题需要找出三个数字之和满足等于零，所以不能使用两数之和的思想，会导致TLE（time limited error）；所以使用两个指针的方法，找出返回答案。
• 由于数字之和为零，所以每组答案中有正数和负数，如果在寻找答案之前将nums排好序，会大大减少时间。
  • 然后开始遍历排序后的数组，这里注意不是遍历到最后一个停止，而是到倒数第三个就可以（因为倒数第二和一的数字不存在 i 和 j）。
  • 剪枝优化：每轮循环开始之处判断当前数字是否与上一轮数字相同，若相同，可跳过；
• 定义两个指针，分别指向fix数字之后开始的数组首尾两个数，
  • 当前三个数拟合的结果大于 0 时，右指针左移；使得指向的数字增大一些；
  • 当前三个数拟合的结果大小于 0 时，左指针右移；使得指向的数字减小一些；
  • 两个指针都需要检测重复数字。

代码实现

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        ans = []
        nums.sort()
		# 遍历到倒数第三个为止
        for k in range(len(nums)-2):
            # 若当前fix数字大于零，则剩余数字全大于0，无法形成有效答案
            if nums[k] > 0:
                break
            # 若本轮fix数字与上一轮一样，则无需重复fix
            if k > 0 and nums[k] == nums[k-1]:
                continue
            # 使用两个指针索引剩余数字
            i, j = k+1, len(nums)-1
            # 循环终止条件
            while i < j:
                # 定义当前和
                s = nums[i] + nums[j] + nums[k]
                # 按照条件，更新相应的指针
                if s < 0:
                    i += 1
                elif s > 0:
                    j -= 1
                else:
                    ans.append([nums[i],nums[j],nums[k]])
                    # 按照条件更新索引值
                    while i < j and nums[i] == nums[i+1]:
                        i += 1
                    while i < j and nums[j] == nums[j-1]:
                        j -= 1
                    # 在下一轮开始之前，更新索引，缩小剩余数组的范围
                    i += 1
                    j -= 1
        return ans