LeetCode 合併兩個有序數組

合併兩個有序數組

---

題目來源：https://leetcode-cn.com/problems/merge-sorted-array/

題目

---

給定兩個有序整數數組 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]

解題思路

---

1. 採用雙指針的方法，從後面往前遍歷
2. 定義 p1 指針指向 num1 數字末尾，p2 指向 num2 數字末尾，p 指向 num1 的最末尾；
3. 從後往前循環遍歷比較兩個數組之間元素的大小；
4. 當 p < 0 時，也就是 p1 和 p2 其中一個小於 0，遍歷結束；
5. 如果 num2 數組有剩餘部分，由於數組都是有序的，直接將 num2 數組剩餘部分放到 num1 數組對應的位置。

代碼實現

---

class Solution:
    def merge(self, nums1, m: int, nums2, n: int) -> None:
        """
        Do not return anything, modify nums1 in-place instead.
        
        合併兩個有序數組爲一個有序數組

        Args:
            num1: 有序數組 1
            m: num1 的元素數量
            num2: 有序數組 2
            n：num2 的元素數量
        
        Returns:
            返回合併後的有序數組
        """
        # 經過雙指針的方法，從後向前遍歷的方法、
        # 定義指針 p1 指向 num1 的數字末尾
        # 定義指針 p2 指向 num2 的數字末尾
        # 定義指針 p 指向 num1 的最末尾
        p1 = m - 1
        p2 = n - 1
        p = m + n - 1
        # 當 p 小於 0，便是 p1 和 p2 其中一個小於 0，遍歷結束
        while p1 >= 0 and p2 >= 0:
            if nums1[p1] < nums2[p2]:
                nums1[p] = nums2[p2]
                p2 -= 1
            else:
                nums1[p] = nums1[p1]
                p1 -= 1
            p -= 1
        # 兩個數組都是有序的
        # 如果 p1 先小於 0，將 num2 剩餘部分放在 num1 數組前面
        nums1[:p2 + 1] = nums2[:p2 + 1]

實現效果

---

---

以上就是本篇的主要內容

---

歡迎關注微信公衆號《書所集錄》