【leetcode】3SUM python实现

最新推荐文章于 2019-06-20 11:46:20 发布

zl87758539

最新推荐文章于 2019-06-20 11:46:20 发布

阅读量2.5k

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/zl87758539/article/details/51578071

版权

leetcode 专栏收录该内容

18 篇文章

订阅专栏

该博客讨论了如何使用Python解决LeetCode中的3SUM问题，即找到数组中所有和为零的不重复三元组。通过先对数组进行排序，然后遍历数组，针对每个元素以它的负值为目标寻找后续的twoSum，实现了O(N^2)的时间复杂度解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.
For example, given array S = {-1 0 1 2 -1 -4},

A solution set is:
(-1, 0, 1)
(-1, -1, 2)

暴力法应该是 $O（N^3）$ ，的时间复杂度，初步估计最少也需要 $O（N^2）$ 的时间复杂度，所以先排序 $(O(Nlog(N)))$ ，从小到大，然后到正数的时候停下。
参考之前的twoSum，相当于是遍历一遍，然后对每个遍历的 $S_i$ ，以 $-S_i$ 为target，往后找twoSum.
代码如下：

class Solution(object):
    def threeSum(self, nums):
        nums = sorted(nums)
        res = []
        if len(nums)<3:
            return []
        for y in xrange(len(nums)):
            if nums[y]>0:
                break
            if y > 0 and nums[y] == nums[y-1]:
                continue
            res = self.twoSum(nums,-nums[y], y, res)
        return res

    def twoSum(self, nums, target, y, res):
        dict={}
        for i in xrange(len(nums)):
            if y >=i:
                continue
            x=nums[i]
            if target-x in dict:
                mlist = sorted([target-x, x, nums[y]])
                if mlist in res:
                    continue
                res.append(mlist)
                continue
            dict[x]=i
        return res