难度:medium
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
个人思路,首先把数组从小到大排序,然后固定一个值使其从左往右遍历,该值右边第一个位置为left,表尾为right
如果三个值相加小于0,说明left的值不够大,向右移动left;相加大于0,说明right的值不够小,向右左移动right——直到left和right相遇
对于排除重复答案,直接用not in判断
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res = []
i = 0
nums.sort(reverse = False)
while i <= len(nums)-3:
left = i+1
right = len(nums) - 1
while left < right:
if nums[i]+nums[left]+nums[right] == 0:
if [nums[i],nums[left],nums[right]] not in res:
res.append([nums[i],nums[left],nums[right]])
left+=1
right-=1
else:
if nums[left]+nums[right]+nums[i] < 0:
left+=1
else: right-=1
i+=1
return res
根据解答区回答修改,思路大致相同,优化有:对于相同的固定值,直接使用continue跳过本次循环,LR也是通过相同思路改变
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res = []
i = 0
nums.sort(reverse = False)
while i <= len(nums)-3:
if i > 0 and nums[i] == nums[i-1]:
i+=1
continue
left = i+1
right = len(nums) - 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[left]+nums[right]+nums[i] < 0:
left+=1
else: right-=1
i+=1
return res