Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.
Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.
简单的归并排序,使用O(n)的额外空间
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int l = 0, r = m, len = m + n;
vector<int> vec;
for (int i = 0; i < len; i++) {
if (l >= m) {
vec.push_back(nums2[r - m]);
r++;
}
else if (r >= len) {
vec.push_back(nums1[l++]);
}
else if (nums1[l] < nums2[r - m]) {
vec.push_back(nums1[l++]);
}
else {
vec.push_back(nums2[r - m]);
r++;
}
}
nums1 = vec;
}
};
网上其它更好的思路:尾插法–>从后向前
这里不需要使用额外的空间
1、如果第一个while循环中,nums1被遍历完毕i < 0,nums1中所有元素都被放入正确的位置,结束循环。第二个while循环接着将nums2剩下的元素依次放入nums1前面的位置。
2、如果第一个while循环中,nums2被遍历完毕j < 0,nums2中所有元素都被放入正确的位置,结束循环。nums1中剩下没有被遍历到的元素,本身就处于正确的位置,无需再考虑。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = m-1, j = n-1, index = m + n - 1;
while( i >= 0 && j >= 0) {
if(nums1[i] > nums2[j]) {
nums1[index--] = nums1[i--];
}else {
nums1[index--] = nums2[j--];
}
}
while( j >= 0) {
nums1[index--] = nums2[j--];
}
}
};
代码进一步优化(不是很直观)
class Solution {
public:
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;
while (k >= 0) {
if (j < 0 || (i >= 0 && nums1[i] > nums2[j]))
nums1[k--] = nums1[i--];
else
nums1[k--] = nums2[j--];
}
}
};
当① j >= 0且i < 0时,② j>=0且nums1[i] < nums2[j]时,执行nums1[k–] = nums2[j–];
当① j<0时,② j >= 0且i >= 0,nums1[i] > nums2[j]时,执行nums1[k–] = nums1[i–];