LeetCode刷題 88 合併兩個有序陣列

2021-10-01 07:30:53 字數 1764 閱讀 8680

給定兩個有序整數陣列 nums1 和 nums2,將 nums2 合併到 nums1 中,使得 num1 成為乙個有序陣列。

說明:

初始化 nums1 和 nums2 的元素數量分別為 m 和 n。

你可以假設 nums1 有足夠的空間(空間大小大於或等於 m + n)來儲存 nums2 中的元素。

雙指標法,遍歷兩陣列,從前往後逐個比較新增,需要輔助空間;

時間複雜度o(m+n),空間複雜度o(m)

class solution(object):

def merge(self, nums1, m, nums2, n):

""":type nums1: list[int]

:type m: int

:type nums2: list[int]

:type n: int

:rtype: none do not return anything, modify nums1 in-place instead.

"""copy_num1 = nums1[:m]

nums1[:] =

# 雙指標

p1 = 0

p2 = 0

while p1 < m and p2 < n:

if copy_num1[p1] < nums2[p2]:

p1 += 1

else:

p2 += 1

if p1 < m:

nums1[p1 + p2:] = copy_num1[p1:]

if p2 < n:

nums1[p1 + p2:] = nums2[p2:]

a = solution()

num1 = [ 1,2,3,0,0,0]

m=3n=3

num2=[2,5,6]

a.merge(num1,m,num2,n)

print(num1)

在方法一上更優化,由題目說明,我們可以在num1陣列中從後往前進行從大至小的新增。不需輔助空間。

class solution(object):

def merge(self, nums1, m, nums2, n):

""":type nums1: list[int]

:type m: int

:type nums2: list[int]

:type n: int

:rtype: none do not return anything, modify nums1 in-place instead.

"""# 雙指標

p1 = m - 1

p2 = n - 1

k = m + n - 1

while p1 >= 0 and p2 >= 0:

if nums1[p1] < nums2[p2]:

nums1[k] = nums2[p2]

p2 -= 1

else:

nums1[k] = nums1[p1]

p1 -= 1

k -= 1

# mnums1[:p2+1] = nums2[:p2+1]

a = solution()

num1 = [ 1,2,3,0,0,0]

m=3n=3

num2=[2,5,6]

a.merge(num1,m,num2,n)

print(num1)

LeetCode刷題之路 88 合併兩個有序陣列

給你兩個有序整數陣列 nums1 和 nums2,請你將 nums2 合併到 nums1 中,使 nums1 成為乙個有序陣列。說明 初始化 nums1 和 nums2 的元素數量分別為 m 和 n 你可以假設 nums1 有足夠的空間 空間大小大於或等於 m n 來儲存 nums2 中的元素。示例...

leetcode刷題 88 合併兩個有序陣列

給你兩個有序整數陣列 nums1 和 nums2,請你將 nums2 合併到 nums1 中 使 nums1 成為乙個有序陣列。這種解法運用到了歸併排序的思想 使用雙指標同時遍歷兩個陣列,比較指標指向元素的大小,將小的元素放到正確的位置即可。class solution def merge self...

LeetCode 題 88 合併兩個有序陣列

給定兩個有序整數陣列 nums1 和 nums2,將 nums2 合併到 nums1 中,使得 num1 成為乙個有序陣列。說明 初始化 nums1 和 nums2 的元素數量分別為 m 和 n。你可以假設 nums1 有足夠的空間 空間大小大於或等於 m n 來儲存 nums2 中的元素。示例 輸...