给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
1、一开始我思维僵化了,遍历nums2找到在nums1中的位置,需要平方复杂度。
2、其实我们可以先合并数组,再排序,复杂度是nlogn。好了一点
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
for (int i = 0; i != n; ++i) {
nums1[m + i] = nums2[i];
}
sort(nums1.begin(), nums1.end());
}
};
3、我门其实忽略了数组有序的条件,我们可以用双指针实现线性复杂度
这一方法将两个数组看作队列,每次从两个数组头部取出比较小的数字放到结果中。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int ans[m+n];
int p1=0;
int p2=0;
int x=0;
while(p1!=m&&p2!=n)
{
if(nums1[p1]<nums2[p2])
{
ans[x]=nums1[p1];
x++;
p1++;
}
else
{
ans[x]=nums2[p2];
x++;
p2++;
}
}
if(p1!=m)
{
while(p1!=m)
{
ans[x]=nums1[p1];
x++;
p1++;
}
}
if(p2!=n)
{
while(p2!=m)
{
ans[x]=nums2[p2];
x++;
p2++;
}
}
for(int i=0;i<m+n;i++)
{
nums1[i]=ans[i];
}
}
};