LeetCode15 三數之和

2022-06-22 05:24:15 字數 1892 閱讀 1124

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