給定乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[ [-1, 0, 1],
[-1, -1, 2]
]
基本思想就是 a+b
+c=0
a+b+c=0
a+b+c=
0 -->−a−
b=
c-a-b=c
−a−b=c
,先計算所有的兩數之和−(a
+b
)-(a+b)
−(a+b)
,然後在所有的和裡面找相等的的數字ccc。
其中有一些修剪的策略:
先排序,使結果公升序並且可以排除一些結果
消除重複:全部找到再去重會超出時間限制,所以自在這裡是使用字典記錄 c 值的index。字典本身查詢快,並且可以消除重複的值,只記錄同乙個值的乙個index(更大的index),然後通過dict[c]>i來避免了重複選擇。
class
solution
:def
threesum
(self, nums)
:'''
list[int]) -> list[list[int]]
'''nums, r =
sorted
(nums)
,set()
for i in
[i for i in
range
(len
(nums)-2
)if i <
1or nums[i]
> nums[i -1]
]:a = nums[i]
dic =
r.update(
[(a, c,
-a - c)
for j, c in
enumerate
(nums[i +1:
])if c in dic and dic[c]
> j]
)return
list
(map
(list
, r)
)
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...