LeetCode 三數之和 python實現

2021-10-04 20:27:32 字數 1295 閱讀 7170

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

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

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

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

[[-1, 0, 1],

[-1, -1, 2]

]在主要思想:

用三個指標k :數指標 , l :左指標, r :右指標,

指標 l 從左往右移動,且始終比 k 大一位,這樣就保證不會跟 k 重疊,

指標 r 從右往左移動,且始終比 l 大,這樣 l 和 r 就不會重疊,即三個指標都不會重疊。

在這個基礎上,實現主要邏輯:

先將數字排序

nums[k]>0時,可以直接跳出迴圈,因為nums[k]都比0大了,後面的nums[i]和nums[j]肯定更大,三者加起來肯定大於0

nums[k]和nums[k-1]相等,前後元素重複,需要過濾

如果nums[l]+nums[r]+nums[k]>0,即三者之和太大了,將 r 指標左移

如果nums[l]+nums[r]+nums[k]<0,說明三者之和太小了,同理將 l 指標右移

如果nums[l]+nums[r]+nums[k]==0,這就是要找的答案,將其儲存起來,同時 l 右移,r左移

注意: l 和 r 在移動的過程中還需要判斷前後元素是否重複

def threesum(nums):

""":type nums: list[int]

:rtype: list[list[int]]

"""nums.sort()

lens = len(nums)

res =

for i in range(lens-2):

k = i

if nums[k] > 0: break

if k>0 and nums[k-1]==nums[k]:continue

l, r = k+1, lens-1

while l0:

r -= 1

while l測試:

threesum([-1, 0, 1, 2, -1, -4])

threesum([-2,0,0,2,2])

threesum([0,0,0])

輸出:

[[-1, -1, 2], [-1, 0, 1]]

[[-2, 0, 2]]

[[0, 0, 0]]

leetcode 三數之和

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

LeetCode 三數之和

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

Leetcode 三數之和

解決方案一 class solution else if nums j nums k target else while i 1 1 i return lllist 解決方案2 class solution 如果有3個0,就直接作為乙個解 if zerosize 3 res.add arrays.a...