问题描述:
问题解答:
(1) 定义两个数组的元素个数 m ,n;
vector<int> nums1 = { 1, 3, 5, 7 };
int m = (int)nums1.size();
vector<int> nums2 = { 2, 4, 6 };
int n = (int)nums2.size();
(2)向数组 nums1 中push_back(0),使得数组nums1中的size为 m + n:
for (int i = 0; i < nums2.size(); i++)
nums1.push_back(0);
(3)设定nums1的后四位数为1,3,5,7
for (int i = n + m - 1; i >= n; i--)
nums1[i] = nums1[i - n];
(4)初始化循环中的变量:
int i = n; // pointer for nums1 [n, n+m)
int j = 0; // pointer for nums2 [0, n)
int k = 0; // pointer merged nums1 [0, n+m)
如图:
(5) 创建循环体while (k < n + m)
①if (nums1[i] < nums2[j]) ,则nums1[k] = nums1[i];
如图:
然后 i++,k++:
② if (nums1[i] > nums2[j]) , 则 nums1[k] = nums2[j]
然后 k++,j++
③ if (j >= n) , 则nums1[k++] = nums1[i++]
④ if (i >= n + m),则nums1[k++] = nums2[j++];
(6)完整的代码实现
#include <iostream>
#include <vector>
#include <cassert>
using namespace std;
void printVec(const vector<int>& vec) {
for (int e : vec)
cout << e << " ";
cout << endl;
}
/// Standard merge process in merge sort
/// Time Complexity: O(n)
/// Space Complexity: O(n)
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
assert(nums1.size() == m + n && nums2.size() == n);
for (int i = n + m - 1; i >= n; i--)
nums1[i] = nums1[i - n]; //输出:nums1[] = {1 3 5 1 3 5 7}
int i = n; // pointer for nums1 [n, n+m)
int j = 0; // pointer for nums2 [0, n)
int k = 0; // pointer merged nums1 [0, n+m)
while (k < n + m) {
if (i >= n + m)
nums1[k++] = nums2[j++];
else if (j >= n)
nums1[k++] = nums1[i++];
else if (nums1[i] < nums2[j])
nums1[k++] = nums1[i++];
else
nums1[k++] = nums2[j++];
}
}
};
int main() {
vector<int> nums1 = { 1, 3, 5, 7 };
int m = (int)nums1.size();
vector<int> nums2 = { 2, 4, 6 };
int n = (int)nums2.size();
for (int i = 0; i < nums2.size(); i++)
nums1.push_back(0); // 输出:nums1[] = {1 3 5 7 0 0 0}.
Solution().merge(nums1, m, nums2, n);
printVec(nums1);
system("pause");
return 0;
}
<全文完>
参考资料: