LeetCode 热题100-6 三数之和

三数之和

给你一个整数数组 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)