思路:
本题适合采用双指针法。
1. 先将数组 nums 升序排序
2. 初始状态:i = 0 ,指针 left = i+1 ,指针 right = len(nums)-1 (left 要 < right)
3. 如果 nums[i] > 0,则不存在满足条件的三元组;
如果 nums[i] + nums[left] + nums[right] > 0,right 指针向前移动;
如果 nums[i] + nums[left] + nums[right] < 0,left 指针向后移动;
如果 nums[i] + nums[left] + nums[right] == 0,找到符合条件的三元组,加入res列表。
4. 去重:
对于 i:当 i > 0 时,如果 nums[i] == nums[i-1],i += 1;
对于 left:如果 nums[left] == nums[left-1],left += 1;
对于 right:如果 nums[right] == nums[right+1],right -= 1.
class Solution(object):
def threeSum(self, nums):
nums.sort()
res=[]
for i in range(len(nums)):
if nums[i]>0:
return res
if i>0 and nums[i]==nums[i-1]:
continue
left,right=i+1,len(nums)-1
while left<right:
sum = nums[i]+nums[left]+nums[right]
if sum>0:
right-=1
elif sum<0:
left+=1
else:
res.append([nums[i],nums[left],nums[right]])
while left<right and nums[right]==nums[right-1]:
right-=1
while left<right and nums[left]==nums[left+1]:
left+=1
left+=1
right-=1
return res