合并两个有序数组
题目描述
给你两个有序整数数组 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]
使用语言: C语言
思路解析: 因为空间复杂度为O(1),使用不能开辟新的空间,给三个下标一个指向nums1(m-1)位置,一个指向nums2(n-1)位置,一个指向nums1(m+n-1)位置,从nums1(m-1)和nums2(n-1)的最后一个元素开始比较,大的放到nums1(m+n-1)位置;
如下图所示:
代码实现
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n)
{
int i = m + n - 1;
int i1 = m - 1;
int i2 = n - 1;
while(i1 >= 0 && i2 >= 0)
{
//从尾开始比较两个数组,哪个大把哪个放到nums1[i]位置
if(nums2[i2] > nums1[i1])
{
nums1[i] = nums2[i2];
--i2;
--i;
}
else
{
nums1[i] = nums1[i1];
--i1;
--i;
}
}
//如果第一个数组先结束,第二个数组没结束直接把第二个拷贝到第一个数组
while(i2 >=0)
{
nums1[i] = nums2[i2];
--i2;
--i;
}
}