leetcode 初级算法
合并两个有序数组
- 思路
额外开辟一个辅助vector result。比较两个数组当前指针位置的首个元素的大小,分三种情况:小于、相等、大于,将两个数组的数字存入另一个数组中。最后,判断是否有一个数组为空而另一个不为空,若成立,则将其剩余元素全部push_back入结果数组result中。
C++代码如下:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
vector<int> result;
int i = 0, j = 0;
while (i < m || j < n) {
while(i < m && j < n) {
if (nums1[i] < nums2[j]) {
result.push_back(nums1[i]);
i++;
}
else if (nums1[i] == nums2[j]) {
result.push_back(nums1[i]);
result.push_back(nums2[j]);
i++;
j++;
}
else {
result.push_back(nums2[j]);
j++;
}
}
if (i >= m && j < n) {
result.push_back(nums2[j]);
j++;
}
if (i < m && j >= n) {
result.push_back(nums1[i]);
i++;
}
}
nums1.assign(result.begin(), result.end());
}
};