給你乙個包含 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
logn)
o(n \log n)
o(nlogn)
,遍歷陣列 o(n
)o\left(n\right)
o(n)
,雙指標遍歷 o(n
)o\left(n\right)
o(n)
,總體 o(n
logn)
+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...