15 三數之和

2021-09-22 12:34:08 字數 1342 閱讀 3368

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

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

例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],

滿足要求的三元組集合為:

[[-1, 0, 1],

[-1, -1, 2]

]定義兩個指標,分別指向fix數字之後開始的陣列首尾兩個數,

class

solution

(object):

defthreesum

(self, nums)

:"""

:type nums: list[int]

:rtype: list[list[int]]

"""ans =

nums.sort(

)# 遍歷到倒數第三個為止

for k in

range

(len

(nums)-2

):# 若當前fix數字大於零,則剩餘數字全大於0,無法形成有效答案

if nums[k]

>0:

break

# 若本輪fix數字與上一輪一樣,則無需重複fix

if k >

0and nums[k]

== nums[k-1]

:continue

# 使用兩個指標索引剩餘數字

i, j = k+1,

len(nums)-1

# 迴圈終止條件

while i < j:

# 定義當前和

s = nums[i]

+ nums[j]

+ nums[k]

# 按照條件,更新相應的指標

if s <0:

i +=

1elif s >0:

j -=

1else

:[nums[i]

,nums[j]

,nums[k]])

# 按照條件更新索引值

while i < j and nums[i]

== nums[i+1]

: i +=

1while i < j and nums[j]

== nums[j-1]

: j -=

1# 在下一輪開始之前,更新索引,縮小剩餘陣列的範圍

i +=

1 j -=

1return ans

15 三數之和

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

15 三數之和

給定乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c 使得 a b c 0 找出所有滿足條件且不重複的三元組。注意 答案中不可以包含重複的三元組。有兩種寫法 第一種耗時久,但容易解讀 class solution if nums.length 3 else if ...

15 三數之和

給定乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c 使得 a b c 0 找出所有滿足條件且不重複的三元組。注意 答案中不可以包含重複的三元組。例如,給定陣列 nums 1,0,1,2,1,4 滿足要求的三元組集合為 1,0,1 1,1,2 分析排序 雙指標避免...