給你乙個包含 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...