三数之和
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
这个题写的太复杂了...用了hash、集合...想法是先把数组排个序,随后用双指针从头开始遍历,然后去hash表中找target是否存在,去重的方式是用集合的方式天然去重(把元素先sorted然后tuple一下扔进集合,因为集合不支持hash列表 - - 字典当然也不支持),虽然做了一部分“剪枝”,但是代码速度也不快。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums.sort()
i,j=0,1
hashmap = {}
hashmap_res =set()
length = len(nums)
for elm in nums:
if elm in hashmap:hashmap[elm]+=1
else:hashmap[elm] = 1
while i<length:
j=i+1
while j<length:
hashmap[nums[i]]-=1
hashmap[nums[j]]-=1
target = 0-(nums[i]+nums[j])
if (nums[j]+nums[i])<=0 and target in hashmap and hashmap[target]!=0:
hashmap_res.add(tuple(sorted([nums[i],nums[j],target])))
hashmap[nums[i]]+=1
hashmap[nums[j]]+=1
j+=1
while j+1<length and nums[j+1]==nums[j]:
j+=1
while i+1<length and nums[i+1]==nums[i]:
i+=1
i+=1
if i<length and nums[i]>0:break
return list(hashmap_res)