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 ) O(n^2) O(n2),其中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))