给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution:
def threeSum(self, nums: [int]) -> [[int]]:
'''
1.排序
2.利用双指针和哨兵寻找点
3.去重,哨兵检测去重;双指针循环去重
'''
n=len(nums)
if (not nums) or n<3:
#剔除空集和小于三位数的情况
return []
res=[]
#先排序,很重要
nums.sort()
print(nums)
for i in range(n):
'''i为哨兵,三元组第一个,之后利用left,right双指针寻找。'''
if(i>0 and nums[i]==nums[i-1]):
#剔除掉,前后哨兵一致的情况
continue
left=i+1
right=n-1
while(left<right):
if (nums[i]+nums[left]+nums[right]==0):
#相同加入
res.append([nums[i],nums[left],nums[right]])
while(left<right and nums[left]==nums[left+1]):
#剔除双指针寻找过程中重复情况
left+=1
while(left<right and nums[right]==nums[right-1]):
right-=1
left+=1
right-=1
elif (nums[i]+nums[left]+nums[right]>0):
#和大于,说明数太大,换小的
right-=1
else:
#和小于,说明数太小,换大的
left+=1
return res