15. 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
排序后双指针 [py]
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
if len(nums) < 3:
return []
nums.sort()
res = []
for i in range(0, len(nums)):
if i > 0 and nums[i] == nums[i-1]:
continue
j = i + 1
k = len(nums) - 1
while j < k:
curr = nums[i] + nums[j] + nums[k]
moveJ = False
moveK = False
if curr == 0:
res.append([nums[i], nums[j], nums[k]])
moveJ = True
moveK = True
elif curr > 0:
moveK = True
else:
moveJ = True
if moveJ:
jv = nums[j]
while j < k and nums[j] == jv:
j += 1
if moveK:
kv = nums[k]
while k > j and nums[k] == kv:
k -= 1
return res