題目

給定兩個(gè)有序整數(shù)數(shù)組 nums1 和 nums2，將 nums2 合并到 nums1 中，使得 num1 成為一個(gè)有序數(shù)組。

說(shuō)明:

初始化 nums1 和 nums2 的元素?cái)?shù)量分別為 m 和 n。
你可以假設(shè) nums1 有足夠的空間（空間大小大于或等于 m + n）來(lái)保存 nums2 中的元素。
示例:

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

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

來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/merge-sorted-array
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

解題思路

本題的核心就是從后往前合并數(shù)組，這樣可以減少移動(dòng)的次數(shù)：

• 從后向前比較數(shù)字的大小，給nums1
• 最后如果nums2還有數(shù)字，直接復(fù)制給nums1

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.
        """
        while m>0 and n>0:
            if nums1[m-1] < nums2[n-1]:
                nums1[m+n-1] = nums2[n-1]
                n = n - 1
            else:
                nums1[m+n-1], nums1[m-1] = nums1[m-1], nums1[m+n-1]
                m = m -1 
        
        if m == 0 and n > 0:
            nums1[:n] = nums2[:n]