三數之和 LeetCode 15

2021-10-24 05:12:31 字數 2176 閱讀 2364

給你乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?請你找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

可以採用排序+雙指標,利用雙指標前後遍歷排序後的陣列,找到和為零的元素加入到空列表中

class

solution

:def

threesum

(self, nums: list[

int])-

> list[list[

int]]:

#若給定的陣列為空,返回空列表

ifnot nums:

return

#對陣列進行排序操作,便於後續前後指標的移動

nums.sort(

)#定義乙個空列表來儲存滿足條件的三元組

res =

n =len(nums)

# 遍歷陣列

for i in

range

(n-2):

#如果陣列前後元素相同,則跳過

if i>

0and nums[i]

== nums[i-1]

:continue

#定義左右指標,分別從兩頭向中間遍歷

l = i +

1 r = n -

1while l < r:

#左指標要始終小於右指標,下面的分析分為了三種情況:和為零,和大於零,和小於零

if nums[i]

+ nums[l]

+ nums[r]==0

:# 滿足條件,加入列表,注意格式

[nums[i]

, nums[l]

, nums[r]])

# 和為零時,繼續查詢,判斷是否有相鄰相同的元素,繼續移動指標

while l < r and nums[l]

== nums[l+1]

: l = l +

1while l < r and nums[r]

== nums[r-1]

: r = r-

1# 查詢所有滿足條件的三元組

l = l +

1 r = r -

1# 大於零時,右指標左移,否則,左指標右移

elif nums[i]

+ nums[l]

+ nums[r]

>0:

r = r -

1else

: l = l +

1# 返回儲存滿足條件三元組的列表

return res

提交結果

時間複雜度:o(n

2)o\left(n^\right)

o(n2

),陣列排序 o(n

log⁡n)

o(n \log n)

o(nlogn)

,遍歷陣列 o(n

)o\left(n\right)

o(n)

,雙指標遍歷 o(n

)o\left(n\right)

o(n)

,總體 o(n

log⁡n)

+o(n

)∗o(

n),o

(n2)

o(n \log n)+o\left(n\right)*o\left(n\right),o\left(n^\right)

o(nlogn)

+o(n

)∗o(

n),o

(n2)

空間複雜度:o(1

)o(1)

o(1)

總感覺好像還是多遍歷了一些東西,比如 i

ii 向後移動,左指標 l

ll 也在向後移動,由於l=i

+1l = i+1

l=i+

1 ,不可避免的會出現多次遍歷的情況,就像遞迴中的重複計算問題,可否有好的解法優化一下?比如加個字典或者備忘錄?

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