题目:
给定两个有序整数数组 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]
解答:
这个题可以考虑一下新开一个m+n那么大的数组,然后不断地将两个数组中间较小的那个放前面,循环下去。
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());
}
};
然而,还有更好的方法:
由于数组num1的空间已经足够,其实是可以不用再浪费空间的。、
那么,可以从数组的最尾部向前一次插入,将比较大的元素放进num1的末尾,最后如果num2有剩余在把num2省的属继续插入,num1有剩余不用管,已经完成了。
下面是代码:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = m - 1, j = n - 1, k = m + n - 1;
while (i >= 0 && j >= 0) {
if (nums1[i] >= nums2[j]) {
nums1[k] = nums1[i];
k--;
i--;
}
else {
nums1[k] = nums2[j];
k--;
j--;
}
}
while (j >= 0) {
nums1[k] = nums2[j];
k--;
j--;
}
}
};
ps:更简洁的代码(来自LeetCode):
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)
{
nums1[index--] = nums1[i] > nums2[j] ? nums1[i--] : nums2[j--];
}
while (j >= 0)
nums1[index--] = nums2[j--];
}
};