LeetCode 15 三數之和 Python 3

2021-10-10 22:40:40 字數 1632 閱讀 4006

1、自己想法,暴力

超時難點:去重

按照題目給的答案來看只能先排序,才能給出與之一樣順序的答案。

class solution:

def threesum(self, nums: list[int]) -> list[list[int]]:

nums.sort()

numslen = len(nums)

target = [[0 for i in range(numslen)] for j in range (numslen)]

result =

for i in range(numslen):

for j in range(i+1, numslen):

target[i][j] = nums[i] + nums[j]

for k in range(j+1, numslen):

flag = 1

if (nums[k] + target[i][j] == 0): #查重

for ind in range(len(result)):

if (nums[k] == result[ind][2]):

if (nums[i] == result[ind][0] or nums[i] == result[ind][1]):

flag = 0

break

if (flag == 1):

index = [nums[i], nums[j], nums[k]]

return result

2、官方解答 排序+雙指標

class solution:

def threesum(self, nums: list[int]) -> list[list[int]]:

nums.sort()

numslen = len(nums)

result=

for k in range(numslen):

if nums[k] > 0: # 之後都大於0,結束迴圈

break

if k>0 and nums[k] == nums[k-1]: # 去重

continue

i, j = k+1, numslen-1

while i < j :

s = nums[k] + nums[i] + nums[j]

if s < 0:

i += 1

while i < j and nums[i] == nums[i-1]: # 跳過所有與當前nums[i]相同的值

i += 1

elif s > 0:

j -= 1

while i < j and nums[j] == nums[j+1]:

j -= 1

else:

temp = [nums[k],nums[i],nums[j]]

i += 1

j -= 1

while i < j and nums[i] == nums[i-1]:

i += 1

while i < j and nums[j] == nums[j+1]:

j -= 1

return result

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...