給定乙個包含 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 分析排序 雙指標避免...