LeedCode15:三数之和

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。

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

解题思路一:

暴力算法(不应该称之为算法):
三层循环找到符合条件的数时间复杂度为 $ O(N^3) $

##解题思路二:
$ c = -(a+b) $ => 申请一个一个set集合将nums中所有的元素存进去,用双层loops很容易找a
和b,再在set集合中便利找到c复杂度为 $ O(N^2) +O(1) = O(N^2)$ ,但是这样会多开辟一块空间。

##解题思路三:
先将整个数组排序python中可以直接用sort()方法进行排序,基于快排时间复杂度为$ O(N\log N)$ 例如nums = [-1, 0, 1, 2, -1, -4]

[-4, -1, -1 , 0, 1, 2 ]

先枚举找打a 一层loop在从剩下的元素中[-1, -1 , 0, 1, 2 ] 找b和c
因a之后的元素是有序的那么我们就可以同时头和尾开始找
假设b从头开始,c从尾开始,如果a+b+c大于零,那么c需要变小c往左移
如果a+b+c大于零,那么b需要变大b往右移

c复杂度为 $O(N^2)$
完整代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
res = []
nums.sort()
for i in range(len(nums)-2):
if i > 0 and nums[i] == 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

最后更新: 2019年03月18日 17:28

原始链接: https://zem12345678.github.io/2019/03/18/LeedCode15:三数之和/

× 请我吃糖~
打赏二维码