题目描述:将两个已排序的数组合并成一个数组
注 意:假设原来的数组1有足够的空间容纳两个数组
思 路:
两个数组分别由后至前比较大小,需要注意的是,vector在使用时,若数组未初始化为一定大小,数组元素的赋值只能使用vec.push_back()完成;若数组已经使用vec.resize(n)初始化大小,才可使用vec[i]的方式对数组进行赋值。
如有需要,请访问我的Github获取包含测试程序的C++源码。
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;
//nums1.resize(k + 1);
while(i >= 0 && j >= 0)
{
if(nums1[i] < nums2[j])
{
nums1[k] = nums2[j];
k--;
j--;
}
else
{
nums1[k] = nums1[i];
k--;
i--;
}
}
while(j >= 0)
{
nums1[k] = nums2[j];
k--;
j--;
}
}
};
20180802更新:不再借助于k变量,思路更简单
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
nums1.resize(m+n);
int i = m - 1;
int j = n - 1;
while(i >= 0 && j >= 0)
{
if(nums1[i] >= nums2[j])
{
nums1[i + j + 1] = nums1[i];
i--;
}
else
{
nums1[i + j + 1] = nums2[j];
j--;
}
}
while(j >= 0)
{
nums1[j] = nums2[j];
j--;
}
}
};