提要
這是一道根據面試題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大,則,如果num1大,則。設置一個最終的計數器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--];
}
}