题目:
给你一个包含 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]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源:https://leetcode-cn.com/problems/3sum
打卡:
这个写的不太好。。时间和空间消耗都太多
import numpy as np
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
if len(nums)<3:
return []
i=0
nums_sorted=sorted(nums)
result_list=[]
for i in range(len(nums_sorted)):
l=i+1
r=len(nums)-1
if nums_sorted[i] > 0:
break
while l<r:
sum=nums_sorted[i]+nums_sorted[l]+nums_sorted[r]
if sum == 0 :
result_list.append([nums_sorted[i],nums_sorted[l],nums_sorted[r]])
l+=1
r-=1
elif sum < 0:
l+=1
elif sum > 0:
r-=1
return np.unique(np.array(result_list),axis=0).tolist()