题目描述
给定一个包含 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