Leetcode.15 三数之和 (采用了双指针算法通过,但是要在集合、元组、列表中来回切换,时间复杂度高,需要优化去重的步骤):
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums=sorted(nums)
if len(nums)<3 or nums[0] > 0:
return []
if nums==[0,0,0]:
return [[0,0,0]]
res=set()
for i in range(len(nums)):
if i-1 > 0 and nums[i] == nums[i-1]:
continue
l = i + 1;r = len(nums) - 1
while l < r:
tmp=[]
if nums[l] + nums[r] == -nums[i]:
tmp.append(nums[l])
tmp.append(nums[r])
tmp.append(nums[i])
l+=1;r-=1
if len(tmp) == 3:
res.add(tuple(tmp))
tmp.clear()
elif nums[l] + nums[r] > -nums[i]:
r-=1
elif nums[l] + nums[r] < -nums[i]:
l+=1
return list(res)