三数之和
https://leetcode-cn.com/problems/3sum/
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res=[]
n=len(nums)
if n<3:
return res
nums.sort()
print(nums)
for i in range(n-2): #遍历nums作为第一个数字
if nums[i]>0: #第一个数字大于0直接退出
break
if i>0 and nums[i-1]==nums[i]: #遍历到相同数字时跳过 不然会重复
continue
left,right = i+1,n-1 #第二、三个数 前后夹击遍历
while left!=right:
# print(nums[i],nums[left],nums[right])
if nums[i]+nums[left]+nums[right]==0:
res.append([nums[i],nums[left],nums[right]])
while left<right: #left后移 若相同字符则继续后移
left+=1
if nums[left-1]!=nums[left]:
break
while left<right: #right前移 若相同字符则继续前移
right-=1
if nums[right+1]!=nums[right]:
break
elif nums[i]+nums[left]+nums[right]>0: #和大于0,要减小
right-=1
else: #和小于0 要增大
left+=1
return res