給你兩個按 非遞減順序 排列的整數數組 nums1 和 nums2,另有兩個整數 m 和 n ,分别表示 nums1 和 nums2 中的元素數目。
請你 合并 nums2 到 nums1 中,使合并後的數組同樣按 非遞減順序 排列。
注意:最終,合并後數組不應由函數傳回,而是存儲在數組 nums1 中。為了應對這種情況,nums1 的初始長度為 m + n,其中前 m 個元素表示應合并的元素,後 n 個元素為 0 ,應忽略。nums2 的長度為 n 。
雙指針法,非最優解,python3 僅供參考學習
nums1 = [1,2,7,0,0,0]
m = 3
nums2 = [2,5,6]
n = 3
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int):
p1 = m - 1
p2 = n - 1
end = m + n - 1
while p2 >= 0:
if nums2[p2] > nums1[p1] or p1 < 0:
nums1[end] = nums2[p2]
p2 -= 1
else:
nums1[end] = nums1[p1]
nums1[p1] = nums2[p2]
p1 -= 1
end -= 1
#print(p1 ,p2, end)
#print('end')
Solution().merge(nums1,m,nums2,n)
print(nums1)