Merge Sorted Array的意思是將兩個有序的數組nums1和nums2合併成爲一個有序的數組nums1。
原題如下:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
}
};
思路如下:對nums1的元素從尾部開始進行移動操作,計算nums1需要移動多少位,i和j分別表示nums1和nums2中沒有歸併的數據數目,則nums1[i-1]和nums2[j-1]分別表示nums1和nums2中剩下的最後一個沒有歸併的數據。假設nums1[i-1]>nums2[j-1],則說明nums[i-1]前面需要插入j個數據,因此將nums1[i-1]後移j位,nums1中沒有歸併的數目減1。同理可知:若nums1[i-1]<=nums2[j-1],表明nums2中的最後一個數在原有位置上應該要增加i個數,移動完之後,nums2中沒有歸併的數目減1。需要注意的一點是,當i=0的時候,nums[i-1]會報錯,因此判斷條件前需要加上i>0。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
for(int i = m,j = n;j > 0;){
if((i > 0) && (nums2[j-1] < nums1[i-1])){
nums1[i-1+j]=nums1[i-1];
i--;
}else{
nums1[i+j-1]=nums2[j-1];
j--;
}
}
}
};