[Leetcode.88][數組]合併兩個有序數組

提要

這是一道根據面試題5:替換空格從後向前遍歷的一道拓展題。

題目

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

思路

思路很清晰,從後向前遍歷,如果num2大,則num2num2--,如果num1大,則num1num1--。設置一個最終的計數器k,將元素放在最終位置。需要注意的是,num1和num2不一定等長,存在一個數組遍歷結束,另外一個沒有結束的情況。循環結束後,需要將剩餘部分拷貝到最終數組中。

 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;
        //邊界判斷
        if(m<=0)
            nums1=nums2;
        while(i>=0&&j>=0)
        {
            if(nums1[i]<nums2[j])
                nums1[k--]=nums2[j--];
            else if(nums1[i]>nums2[j])
                nums1[k--]=nums1[i--];
            else
                nums1[k--]=nums2[j--];
        }
        while(i>=0)
        {
            nums1[k--]=nums1[i--];
        }
        while(j>=0)
        {
            nums1[k--]=nums2[j--];
        }
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章