Leetcode 三數之和

2021-10-12 10:35:31 字數 1506 閱讀 5301

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

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

示例:

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

滿足要求的三元組集合為: [ [-1, 0, 1], [-1, -1, 2] ]

雙指標即可:

class

solution

:def

threesum

(self, nums: list[

int])-

> list[list[

int]]:

n =len(nums)

nums.sort(

) ans =

list()

# 剪枝

if n <3:

return ans

for i in

range

(n-2):

#去除重複

if i >

0and nums[i]

== nums[i-1]

:continue

#剪枝:因為是排序之後的,如果nums[i] + nums[i+1] + nums[i+2] > 0說明後面的也全會大於0,跳出迴圈,迴圈結束。nums[i] + sum(nums[-1:-3:-1])<0的話,當前迴圈結束,進入下乙個迴圈。

#這裡的剪枝是關鍵,可以極大的排除很多因素

if nums[i]

+ nums[i+1]

+ nums[i+2]

>0:

break

if nums[i]

+sum

(nums[-1

:-3:

-1])

<0:

continue

#雙指標

l, r = i+

1, n-

1while l < r:

tmp = nums[i]

+ nums[l]

+ nums[r]

if tmp ==0:

[nums[i]

, nums[l]

, nums[r]])

r -=

1 l +=

1#這裡是去除l和r位置重複的元素

while l < r and nums[r]

== nums[r+1]

:r -=

1while l < r and nums[l]

== nums[l-1]

:l +=

1elif tmp >0:

r -=

1continue

else

: l +=

1continue

return ans

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