R7-双指针篇
思路:
三个元素,代表需要3个指针,利用双指针收缩的思想,我们可以设置k,i,j三个元素指针。
k代表最外层的循环,循环一遍数组。(为了降低时间复杂度以及搜索难度,我们先将nums sort一下)
i,j代表双指针收缩,i从k+1开始,j从n-1往回收。
tips:
本题一大重点在于重复值的处理上。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
ret=[]
nums.sort()
n=len(nums)
for k in range(n-2):
if nums[k]>0:
break
#重复跳过
if k>0 and nums[k]==nums[k-1]:
continue
i=k+1
j=n-1
#同一个k的不同i,j组合
while i<j:
sum=nums[k]+nums[i]+nums[j]
if sum<0:
#扩展i
i+=1
while i<j and nums[i]==nums[i-1]:
i+=1
elif sum>0:
#收缩j
j-=1
while i<j and nums[j]==nums[j+1]:
j-=1
else:
ret.append([nums[k],nums[i],nums[j]])
i+=1
j-=1
#还是要处理重复
while i<j and nums[i]==nums[i-1]:
i+=1
while i<j and nums[j]==nums[j+1]:
j-=1
return ret