&LeetCode88& 合併兩個有序數組

題目

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

來源:力扣(LeetCode

思路

本題中,可以從 nums1 和 nums2 數組的末尾開始逐個比較,將較大的數,從後往前加入混合數組的末尾;
首先,需要三個變量 i,j,k,分別指向 nums1,nums2,和混合數組的末尾;
其次,進行 while 循環,若i和j都大於0,再看如果 nums1[i] > nums2[j],說明要先把 nums1[i] 加入混合數組的末尾,加入後k和i都要自減1;
再次,若nums1[i] < nums2[j],就把 nums2[j] 加入混合數組的末尾,加入後k和j都要自減1;
循環結束後,可能i或者j還大於等於0,若j大於0,那麼還需要繼續循環,將 nums2 中的數字繼續拷入 nums1中;
若是i大於等於0,不用管,因爲混合數組本身就放在 nums1 中的。

C++代碼

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--];
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章