C++實現LeetCode(88.混合插入有序數組)

Posted on by

[LeetCode] 88. Merge Sorted Array 混合插入有序數組

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:

  • The number of elements initialized in nums1and nums2 are m and n respectively.
  • You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2.

Example:

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

Output: [1,2,2,3,5,6]

混合插入有序數組，由於兩個數組都是有序的，所有隻要按順序比較大小即可。題目中說瞭 nums1 數組有足夠大的空間，說明不用 resize 數組，又給瞭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, j = n - 1, 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, j = n - 1, k = m + n - 1;
        while (j >= 0) {
            nums1[k--] = (i >= 0 && nums1[i] > nums2[j]) ? nums1[i--] : nums2[j--];
        }
    }
};

到此這篇關於C++實現LeetCode(88.混合插入有序數組)的文章就介紹到這瞭,更多相關C++實現混合插入有序數組內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：