(1)15.三数之和
题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
解题思路:
双指针 定一寻二
剪枝
1使用排序后第一个值来确定,最小值大于0则结束
2为避免出现重复情况
while left < right and nums[left] == nums[left+1]
while left < right and nums[right] == nums[right-1]
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
L = len(nums)
if L <= 2:
return []
res = []
nums.sort()
for i in range(L):
if nums[i] > 0:
break
if i > 0 and nums[i] == nums[i-1]:
continue
left = i + 1
right = L - 1
while left < right:
tempSum = nums[i] + nums[left] + nums[right]
if tempSum == 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 tempSum < 0:
left += 1
elif tempSum > 0:
right -= 1
else:
pass
return res