用的是O(mn)的方法,稍微优化了下。
依次遍历nums2,并取值与nums1中的值比较,nums1每次起始地址为index,
每次插入一个新的值后,nums1的起始地址index更新为插入数字所指示的下标。
代码:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int index=0;
for(int i=0;i<n;i++)
{
int flag=0;
for(int j=index;j<m;j++)
{
if(nums2[i]<nums1[j])
{
for(int k=m;k>j;k--) nums1[k]=nums1[k-1];
nums1[j]=nums2[i];
index=j+1;
m++;
flag=1;
break;
}
}
if(flag==1) continue;
nums1[m]=nums2[i];
index=m;
m++;
}
}
};
有一个O(m+n)的方法:摘自链接:https://www.jianshu.com/p/88203168eb42
提示中已经给出,假设array1有足够的空间了,于是我们不需要额外构造一个数组,并且可以从后面不断地比较元素进行合并。
- 比较array2与array1中最后面的那个元素,把最大的插入第m+n位
- 改变数组的索引,再次进行上面的比较,把最大的元素插入到array1中的第m+n-1位。
- 循环一直到结束。循环结束条件:当index1或index2有一个小于0时,此时就可以结束循环了。如果index2小于0,说明目的达到了。如果index1小于0,就把array2中剩下的前面的元素都复制到array1中去就行。