给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路:第一种暴力;第二种用set查找;第三种排序,可以不用set,空间复杂度优化,但是索引会改变。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
if len(nums) < 3:
return []
nums.sort()
res = set()
for i,v in enumerate(nums[:-2]):
if i >=1 and v == nums[i-1]:
continue
d = {}
for x in nums[i+1:]:
if x not in d:
d[-v-x] = 1
else:
res.add((v,-v-x,x))
#return map(list,res) #python2环境下的
return list(map(list,res)) #python3环境下的