LeetCode题解(python)-15. 三数之和

LeetCode题解（python）

15. 三数之和

题目描述

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

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

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

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

解题心得

看到这种处理数组中某几个元素的关系的题，首先想到的是遍历，但是单纯三个元素遍历的时间复杂度很高，很有可能超时；其次想到的是双指针遍历（往中间夹，当然还有一种是从中间往两边走，思想相似）

在实际解决过程中，有几个小tip：

• 加一个判断，跳过循环，可以提升接近3倍的速度，可能系统给出的时间有偏差，与测试用例有关。

  • if i!=0 and numsort[i] == numsort[i-1]:
                    continue
    

    避免了重复的计算判别不可能的情况

• 判断的时候，两边的指针根据当前元素关系的偏向进行移动

• 最后得到的列表中有重复的元素，使用set去除列表中重复的元素

  • list(set(rel))
    

  • 这里注意，前面rel里面append元素时要用（key1,key2,key3）

• 时间复杂度：$O(n^2)$，其中n为列表长度

解题代码

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        
        nums.sort()
        numsort = nums
        rel = []
        for i in range(len(numsort)):
            j = i+1
            k = len(numsort) -1 
            if i!=0 and numsort[i] == numsort[i-1]:
                continue
            while j<k:
                if (numsort[j]+numsort[k]+numsort[i]) == 0:
                    rel.append((numsort[i],numsort[j],numsort[k]))
                    j = j+1
                    k = k-1
                elif (numsort[j]+numsort[k]+numsort[i]) < 0:
                    j = j+1
                else:
                    k = k-1
        return list(set(rel))