题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
排序,双指针:首先对数组进行排序,遍历数组中的每个数作为三元组中的一个数i,应用双指针获取当前数i之后的另外两个数。
细节:如何避免重复解?
算法流程
- 特殊情况,数组长度 n,如果数组为 null 或者数组长度小于 3,返回 []。
- 对数组进行排序。
- 遍历排序后数组中的每个数作为三元组中的一个数:
- 若 nums[i]>0:因为已经排序好,所以后面的数越来越大,不可能再有两个数与其之和等于 0,直接跳出遍历,返回结果
- 遇到重复元素:跳过,避免重复解
- 应用双指针获取另外两个数,令左指针 L=i+1(从左往右走),右指针 R=n−1(从右往左走),当 L<R 时,执行循环:
- 当nums[i]+nums[L]+nums[R]==0,把当前三元组放入结果,继续执行循环,同时将 L,R移到下一位置,寻找新的解(若左界和右界和下一位置重复则跳过,移到下下位置,以此类推)
- 若三数之和大于 0,说明 nums[R] 太大,R 左移
- 若三数之和小于 0,说明 nums[L]太小,L 右移
代码
class Solution:
def three_sum(self, nums: List[int]) -> List[List[int]]:
n=len(nums)#数组长度
if(not nums or n<3):
return []#特殊情况返回
res=[]#存储结果
nums.sort()#排序
#遍历
for i in range(n):
if(nums[i]>0):
return res
if(i>0 and nums[i]==nums[i-1]):
continue#避免重复解
#双指针
L=i+1
R=n-1
while(L<R):
if(nums[i]+nums[L]+nums[R]==0):
res.append([nums[i],nums[L],nums[R]])
#避免重复解
while(L<R and nums[L]==nums[L+1]):
L=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
L=L+1
R=R-1
elif(nums[i]+nums[L]+nums[R]>0):
R=R-1
else:
L=L+1
return res