Leetcode 88. 合併兩個有序數組（python題解）

1 題目

給你兩個有序整數數組 nums1 和 nums2，請你將 nums2 合併到 nums1 中，使 nums1 成爲一個有序數組。

說明:
初始化 nums1 和 nums2 的元素數量分別爲 m 和 n 。
你可以假設 nums1 有足夠的空間（空間大小大於或等於 m + n）來保存 nums2 中的元素。

2 示例

輸入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

輸出: [1,2,2,3,5,6]

3 分析

要將 nums2 合併到 nums1 中，如果從前向後合併，那麼合併後的值可能會覆蓋還未進行比較的值，又因爲 nums1 有足夠的空間，所以我們選擇從後向前合併。

4 題解

class Solution:
	def merge(self, nums1, m, nums2, n):
		p1 = m-1  # 遍歷nums1包含元素的部分
		p2 = n-1  # 遍歷nums2
		p = m+n-1 # 遍歷nums1用於合併
		while p1 >= 0 or p2 >= 0:
			if p1 < 0:  # nums1包含元素的部分遍歷完
				nums1[p] = nums2[p2]
				p2 -= 1
			elif p2 < 0:  # nums2遍歷完
				nums1[p] = nums1[p1]
				p1 -= 1
			elif nums1[p1] > nums2[p2]:
				nums1[p] = nums1[p1]
				p1 -= 1
			else:
				nums1[p] = nums2[p2]
				p2 -= 1
			p -= 1