难度就是结果进行去重。
之前做过一道两数之和的题,本题也采用双指针解法。
特判:当数组长度小于3时,返回空列表
对数组进行排序,方便之后的去重
对数组进行遍历,当元素,与前边的相同时,跳过本次循环,当数组的元素大于0时,之后的元素肯定没有可以相加为0的三个数了。因此返回结果
假设l = i - 1 , r = len(nums) -1,如果nums[i]+nums[l]+nums[r]>0,就调整r,反之调整l,在调整l和r的过程中也要多l,和r对应的值,进行去重。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
if len(nums) < 3:
return []
n = len(nums)
nums.sort()
print(nums)
res = []
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:
print(l,r)
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:
l = l + 1
elif nums[i] + nums[l] + nums[r] > 0:
r = r - 1
return res