給定乙個包含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]]:
res =
nums.sort(
)# 首先需要排序
for i in
range
(len
(nums)-2
):if i >
0and nums[i]
== nums[i-1]
:# 去除重複情況, 如[0 , 0 , 0, 0]
continue
left, right = i +1,
len(nums)-1
while left < right:
cur = nums[i]
+ nums[left]
+ nums[right]
if cur ==0:
[nums[i]
, nums[left]
, nums[right]])
# 去除重複情況, 如[-2, 0, 0, 2, 2]
while left < right and nums[left]
== nums[left+1]
: left +=
1while left < right and nums[right]
== nums[right-1]
: right -=
1# 相等之後 left right指標各移動一步
left +=
1 right -=
1elif cur <0:
left +=
1else
: right -=
1return res
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...