給定乙個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[ [-1, 0, 1],
[-1, -1, 2]
]
這種數值列表中有加減的題目,排序後可能會更加方便些。
固定兩個數,將 target 設定為兩個數之和的相反數,看 target 是否在列表中,若在,則為乙個結果。
class solution:
def threesum(self, nums):
""":type nums: list[int]
:rtype: list[list[int]]
"""ret =
# 陣列題,其中有加減操作的,排序後會方便很多
nums.sort()
len_nums = len(nums)
for i in range(len_nums - 1):
for j in range(i+1, len_nums):
# 固定下來兩個數,如果另外乙個數也在陣列中,則為乙個結果
if -(nums[i]+nums[j]) in nums[j+1:]:
a_ret = [nums[i], nums[j], -(nums[i]+nums[j])]
if a_ret not in ret:
return ret
但是這種方法有兩層迴圈,果然報了 tle(time limit exceeded) 錯誤。
固定乙個數,設定 target 為其他兩個數之和的相反數,即 0 - 當前數,由兩頭向中間遍歷陣列,找到之和符合 target 的兩個數即是乙個結果。
class solution:
def threesum(self, nums):
""":type nums: list[int]
:rtype: list[list[int]]
"""ret =
nums.sort()
for i in range(0, len(nums)):
# 若存在相同數,跳過以減少計算
if i > 0 and nums[i] == nums[i-1]:
continue
target = 0 - nums[i]
start, end = i+1, len(nums)-1
while start < end:
# 頭尾之和大於目標值,尾向前移一位
if nums[start] + nums[end] > target:
end -= 1
# 頭尾之和小於目標值,頭向後移一位
elif nums[start] + nums[end] < target:
start += 1
# 頭尾之和等於目標值,是乙個結果,新增進去
# 頭尾都移動一位
else:
start += 1
end -= 1
# 若移動後有重複值,繼續移動
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...