美文网首页
LeetCode 15 三数之和 3Sum Python

LeetCode 15 三数之和 3Sum Python

作者: 划水型派大星 | 来源:发表于2019-05-08 10:28 被阅读0次

    有关哈希表的LeetCode做题笔记,Python实现

    15. 三数之和 3Sum

    LeetCodeCN 第15题链接

    第一种方法:三重遍历,时间复杂度为O(n^3)

    第二种方法:两重遍历得到前两个数,然后查询第三个数-(a+b)是否存在。用哈希表set()

    class Solution(object):
        def threeSum(self, nums):
            """
            :type nums: List[int]
            :rtype: 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)
    
    

    第三种方法:先升序排序,一遍遍历,然后在后面的新数组里用双指针检查三个数之和是否为0,大于0则右指针向左走,小于0则左指针向右走。

    class Solution(object):
        def threeSum(self, nums):
            if len(nums) < 3:
                return []
            nums.sort()
            res = []
            for i, x in enumerate(nums[:-2]):
                if i >= 1 and x == nums[i-1]:
                    continue
                l, r = i+1, len(nums)-1
                while l < r:
                    s = nums[i] + nums[l] + nums[r]
                    if s < 0:
                        l += 1
                    elif s > 0:
                        r -= 1
                    else:
                        res.append((nums[i], nums[l], nums[r]))
                        while l < r and nums[l] == nums[l+1]:
                            l += 1
                        while l < r and nums[r] == nums[r-1]:
                            r -= 1
                        l += 1
                        r -= 1
            return res
    

    上一题:1. 两数之和 Two Sum

    相关文章

      网友评论

          本文标题:LeetCode 15 三数之和 3Sum Python

          本文链接:https://www.haomeiwen.com/subject/sdapoqtx.html