给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
之前有做过一次,简单方法的https://mp.csdn.net/mp_blog/creation/editor/121461889
但既然又来啦,那就是有第二种方法咯:
将第二个数组nums2中的元素放入数组nums1中,因为事先,nums1中预留了位置给nums2,元素为0。放得下。
又因为两个数组都是非递减顺序,即元素在增加,但可能会有重复元素。所以我们可以再创建一个数组nums3[m+n],每次分别那两个数组中最小的那个数出来比较,把较小的那个数尾插入nums3中,最后在把nums3中的所有元素,挨个赋值给nums1.
我们需要考虑到一些情况,倘若nums1中元素为0或者元素已经全部插入nums3中,那么我们只需要将nums2中的元素依次放入nums3。反之,倘若nums2中元素个数为0或者元素已经全部插入nums3,那么我们只需将nums1中的元素依次放入nums3中。
此时,考虑正常情况,那就是比大小嘛,谁小就放谁进nums3。
最后,我们再把nums3中的元素赋值给nums1.
代码如下:
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i=0,j=0;
int temp;
int nums3[m+n];
while(i<m||j<n)
{
if(i==m)
{
temp=nums2[j++];
}
else if(j==n)
{
temp=nums1[i++];
}
else if(nums1[i]<nums2[j])
{
temp=nums1[i++];
}
else
{
temp=nums2[j++];
}
nums3[i+j-1]=temp;
}
for(int i=0;i<m+n;i++)
{
nums1[i]=nums3[i];
}
}
};