给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
CODE:
class Solution:
def threeSum(self, nums):
nums.sort() # 先排序
n = len(nums)
left = 0
right = n - 1
results = []
# 固定一个数,另外2个数从两边夹,3个数的和大于0,右边最大值就减少,如果小于0,左右最大值就增加
# 固定数如果相同的话(因为排序位置会在一起),结果可能会相同,所以直接跳过
for index in range(0, n):
if index >= 1 and nums[index] == nums[index - 1]:
continue
# 左边从固定数右边最小值开始,右边从最大值开始
left, right = index + 1, n - 1
while left < right:
total = nums[left] + nums[index] + nums[right]
if total == 0:
results.append([nums[index], nums[left], nums[right]])
# 如果和为0,左右各前进一步,如果只变left,right其中一个,由于left,index固定,3个数和固定,那么right肯定固定,所以左右都要变
left += 1
right -= 1
# 左右前进的时候,如果前进后的值和原值相同则需要继续前进,不然会重复
# 如:[-2, 0, 0, 2, 2],a: index(0,1,4), b:index(0, 2, 3)都是[-2,0,2]会重复
while left < right and nums[left] == nums[left - 1]:
left += 1
while left < right and nums[right] == nums[right + 1]:
right -= 1
elif total > 0:
right -= 1
else:
left += 1
return results
su = Solution()
nums = [-2, 0, 0, 2, 2]
res = su.threeSum(nums)
print(res)