天天看點

leetcode之Merge Sorted Array(88)

題目:

給定兩個有序整數數組 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成為一個有序數組。

說明:

  • 初始化 nums1 和 nums2 的元素數量分别為 m 和 n。
  • 你可以假設 nums1 有足夠的空間(空間大小大于或等于 m + n)來儲存 nums2 中的元素。

示例:

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

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

python代碼1:

class Solution:
    def merge(self, nums1, m, nums2, n):
        p, q, k = m-1, n-1, m+n-1
        while p >= 0 and q >= 0:
            if nums1[p] > nums2[q]:
                nums1[k] = nums1[p]
                p, k = p-1, k-1
            else:
                nums1[k] = nums2[q]
                q, k = q-1, k-1
        nums1[:q+1] = nums2[:q+1]
        
           

python代碼2:

class Solution:
    def merge(self, nums1, m, nums2, n):
        nums1[m:m+n]=nums2[:n]
        nums1.sort()
           

心得:自己的代碼冗長,網上的代碼精簡,不多說了,好好研究一波。代碼一是我們應該需要掌握的,代碼二有點開挂了,這裡并不推薦。