day_19 —— LeetCode 88：合併兩個有序數組

Day_19 —— $LeetCode 88$：合併兩個有序數組

題目

由於兩個數組都是有序的，所有隻要按順序比較大小即可。題目中說了 nums1 數組有足夠大的空間，又給了m和n，那就知道了混合之後的數組的大小，這樣就從 nums1 和 nums2 數組的末尾開始一個一個比較，把較大的數，按順序從後往前加入混合之後的數組末尾。

設置三個變量 i，j，k，分別指向 nums1，nums2，和混合數組的末尾。進行 while 循環，如果i和j都大於0，再看如果 nums1[i] > nums2[j]，說明要先把 nums1[i] 加入混合數組的末尾，加入後k和i都要自減1；反之就把 nums2[j] 加入混合數組的末尾，加入後k和j都要自減1。循環結束後，有可能i或者j還大於等於0，若j大於0，那麼還需要繼續循環，將 nums2 中的數字繼續拷入 nums1。若是i大於等於0，那麼就不用管，因爲混合數組本身就放在 nums1 中。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int i = m-1; 
        int j = n-1;
        int k = m+n-1;

        while (i>=0 && j>=0){
            if (nums1[i] > nums2[j])
                nums1[k--] = nums1[i--];
            else 
                nums1[k--] = nums2[j--];
        }
        while (j>=0){
            nums1[k--] = nums2[j--];
        }
    }
};

我們還可以寫的更簡潔一些，將兩個 while 循環融合到一起，只要加上 i>=0 且 nums1[i] > nums2[j] 的判斷條件，就可以從 nums1 中取數，否則就一直從 nums2 中取數。

class Solution{
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n){
        int i = m-1;
        int j = n-1;
        int k = m+n-1;

        while(j >= 0){
            nums1[k--] = i >= 0 && nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
        }
    }

};

線下完整代碼實現：

#include <iostream>
#include <vector>
using namespace std;

class Solution 
{
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) 
    {
        int i = m-1;
        int j = n-1;
        int k = m+n-1;

        while(j >= 0){
            nums1[k--] = i >= 0 && nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
        }
    }
};

int main()
{
    vector<int> nums1 = {1,2,3,0,0,0};
    vector<int> nums2 = {2,5,6};

    int m = 3;
    int n = 3;
    Solution().merge(nums1, m, nums2, n);

    for (auto& num : nums1)
    {
        cout << num << " ";
    }
    cout << endl;
    return 0;
}