给你一个包含 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]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
最先想到的应该就是for循环+双指针,然后去重
第一次错误写法
def threeSum(nums):
n = len(nums)
nums.sort()
if n < 3:
return []
arr = []
for i in range(n):
if nums[0] > 0:
return []
left = i + 1
right = n - 1
if i > 0 and nums[i] == nums[i-1]:
continue
while left < right:
target = nums[i] + nums[left] + nums[right]
if target == 0:
arr.append([nums[i],nums[left],nums[right]])
left += 1
right -= 1
elif target > 0:
right -= 1
while left < right and nums[right] == nums[right + 1]:
right -= 1
else:
left += 1
while left < right and nums[left] == nums[left - 1]:
left += 1
return arr
刚开始我想的是,在等于0时去重跟else里面去重一样,但是不一样,会增加重复的列表
例如:nums = [-2,0,0,2,2],所以也要在这里去重,在else里面增加去重,应该是减少了时间复杂度,让程序执行得更快。
第二个错误
while left < right and nums[right] == nums[right + 1]:
我写成了nums[right - 1],当遇到了[0,0,0,],就会跳过去。
再次明确下,这个去重是当前已经出现了相同元素,这次应该跳过这个元素,不使用它了
完整版
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n =len(nums)
nums.sort()
if n < 3:
return []
arr = []
for i in range(n):
if nums[0] > 0:
return []
left = i + 1
right = n - 1
if i > 0 and nums[i] == nums[i-1]:
continue
while left < right:
target = nums[i] + nums[left] + nums[right]
if target == 0:
arr.append([nums[i],nums[left],nums[right]])
left += 1
while left < right and nums[left] == nums[left - 1]:
left += 1
right -= 1
while left < right and nums[right] == nums[right + 1]:
right -= 1
elif target > 0:
right -= 1
while left < right and nums[right] == nums[right + 1]:
right -= 1
else:
left += 1
while left < right and nums[left] == nums[left - 1]:
left += 1
return arr