题目 :给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums.sort()
count = len(nums)
collect = []
for i in range(count):
left = i+1
right = count-1
#避免重复找同一个数据
#排序后数组为[-4,-1,-1,0,1,2] 当i为第一个-1的时候,已经查找到[-1, 0, 1],当碰到第二个-1时,避免再判断一次,多加一遍[-1, 0, 1],要退出本次循环,开始下一轮
if i >0 and nums[i] == nums[i-1]:
left +=1
continue
#数据按小到大排列,每次选取nums[i],在其后寻找符合a + b + c = 0的两个数据
while left < right:
sum = nums[i] + nums[left] + nums[right]
if sum == 0:
col = [nums[i],nums[left],nums[right]]
collect.append(col)
left+=1
right-=1
#循环中nums[i]已确定,当再确定1个数时,另一个数也确定,左右端任一端出现重复均可跳过
while nums[left] == nums[left-1] and left < right:
left+=1
while nums[right] == nums[right+1] and left < right:
right-=1
if sum<0:
left+=1
elif sum > 0:
right-=1
return collect