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
解题思路
双指针+去重
大框架就是:先从小到大排序,第一个层循环用来遍历,里边一对双指针从两侧遍历。
然后就是细节处理,主要包括:去重,双指针的遍历终止。
详细可以看代码,都有注释。
代码
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums.sort()#对数据进行排序
res=list()#初始化返回列表
for first in range(len(nums)):#第一层循环,选择第一个数
if first>0 and nums[first]==nums[first-1]:#去重(若当前数跟前一个数一样则忽略,第0个数没有前一个数)
continue
third=len(nums)-1#指向尾巴,与第二个指针形成双指针,加快查找速度
for second in range(first+1,len(nums)):#第二个指针,从第一个的后一个开始,避免重复
if second>first+1 and nums[second]==nums[second-1]:#与第一个指针同理,去重
continue
while second<third and nums[first]+nums[second]+nums[third]>0:#在第二到第三个指针之间遍历两个数
third-=1
if second==third:#若两个指针相交,则说明没有满足的,就退出
break
if nums[first]+nums[second]+nums[third]==0:#若两个指针不相交,三个数相加等于0 则保存起来
res.append([nums[first],nums[second],nums[third]])
return res