leetcode-腾讯精选练习（50 题）python #15 三数之和

最新推荐文章于 2019-06-27 19:07:12 发布

楚天星云

最新推荐文章于 2019-06-27 19:07:12 发布

阅读量137

点赞数

分类专栏： leetcode-腾讯精选练习50题

本文链接：https://blog.csdn.net/weixin_43866606/article/details/91356867

版权

leetcode-腾讯精选练习50题专栏收录该内容

49 篇文章 1 订阅

订阅专栏

leetcode-腾讯精选练习（50 题）python #15 三数之和
题目来源：https://leetcode-cn.com/problemset/50/
给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums, r = sorted(nums), set()
        for i in [i for i in range(len(nums)-2) if i < 1 or nums[i] > nums[i-1]]:
            d = {-nums[i]-n: j for j, n in enumerate(nums[i + 1:])}
            r.update([(nums[i], n, -nums[i]-n) for j, n in enumerate(nums[i+1:]) if n in d and d[n] > j])
        return list(map(list, r))