题目:给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。
用双指针可以快速解决问题,注意去重即可,重复内容有两部分:锚点i重复和指针部分重复
def threeSum(num ):
res=[]
if len(num)<3:
return res
num.sort()
for i in range(len(num)-2):
if i != 0 and num[i] == num[i - 1]:
continue
left=i+1
right=len(num)-1
while(left<right):
if num[left]+num[right]==-num[i]:
res.append([num[i],num[left],num[right]])
while left + 1 < right and num[left] == num[left + 1]:
left += 1
while right - 1 > left and num[right] == num[right - 1]:
right -= 1
left += 1
right-= 1
elif num[left]+num[right]<-num[i]:
left+=1
elif num[left] + num[right] > -num[i]:
right-=1
return res