找工作刷题记录_012三数之和

原创于 2019-03-23 11:03:53 发布 · 181 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#找工作刷题

本文介绍了一种寻找数组中三个数相加为零的所有不重复组合的方法。通过排序和使用哈希表来优化搜索过程，避免了重复解的问题，并将时间复杂度降低到O(n^2)。

给定一个包含 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环境下的