Leetcode-15.三数之和

题目描述

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

题目分析

基本思想就是 $a+b+c=0$ -->$-a-b=c$ ，先计算所有的两数之和$-(a+b)$，然后在所有的和里面找相等的的数字$c$。

其中有一些修剪的策略：

1. 先排序，使结果升序并且可以排除一些结果
2. 消除重复：全部找到再去重会超出时间限制，所以自在这里是使用字典记录 c 值的index。字典本身查询快，并且可以消除重复的值，只记录同一个值的一个index（更大的index），然后通过dict[c]>i来避免了重复选择。

代码示例

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]]:
            a = nums[i]
            dic = {-a - b: j for j, b in enumerate(nums[i + 1:])}
            r.update([(a, c, -a - c)
                      for j, c in enumerate(nums[i + 1:]) if c in dic and dic[c] > j])
        return list(map(list, r))