给你一个包含 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
res = []
#先做排序
nums.sort()
for a in range(len(nums)):
if nums[a]>0:
return res
#重复元素直接跳过,确保第一个元素不被跳过
if a>0 and nums[a] ==nums[a-1]:
continue
#将b和c初始化,固定a以后b和c一个从前往后一个从后往前寻找可能的答案
b = a+1
c = len(nums)-1
while b<c:
if nums[a]+nums[b]+nums[c] == 0:
res.append([nums[a],nums[b],nums[c]])
#去重
while b<c and nums[b] == nums[b+1]:
b+=1
while b<c and nums[c] == nums[c-1]:
c-=1
b+=1
c-=1
#大于0时c-1
elif nums[a]+nums[b]+nums[c]>0:
c-=1
#小于0时b+1
else:
b+=1
return res