Leetcode 15 三數之和

2021-09-22 05:52:40 字數 1227 閱讀 9438

給定乙個包含 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=0

a+b+c=

0 -->−a−

b=

c-a-b=c

−a−b=c

,先計算所有的兩數之和−(a

+b

)-(a+b)

−(a+b)

,然後在所有的和裡面找相等的的數字ccc。

其中有一些修剪的策略:

先排序,使結果公升序並且可以排除一些結果

消除重複:全部找到再去重會超出時間限制,所以自在這裡是使用字典記錄 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 <

1or nums[i]

> nums[i -1]

]:a = nums[i]

dic =

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)

)

LeetCode 15 三數之和

15.給定乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c 使得 a b c 0 找出所有滿足條件且不重複的三元組。注意 答案中不可以包含重複的三元組 方法一,個人解法正確,但是效率太低,時間複雜度o n 3 時間超時,無法提交至leetcode public s...

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 class solutio...

leetcode15 三數之和

給定乙個包含 n 個整數的陣列nums,判斷nums中是否存在三個元素 a,b,c 使得 a b c 0 找出所有滿足條件且不重複的三元組。注意 答案中不可以包含重複的三元組。例如,給定陣列 nums 1,0,1,2,1,4 滿足要求的三元組集合為 1,0,1 1,1,2 先找兩數之和,然後再用un...