88. 合并两个有序数组
给你两个有序整数数组 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]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
提示:
nums1.length == m + n
nums2.length == n
0 <= m, n <= 200
1 <= m + n <= 200
-109 <= nums1[i], nums2[i] <= 109
题解:
很明显我们直接采用双指针原理即可完成合并,即比对两个数组的一个数字,谁大取谁指针挪移即可。但是从头遍历的话会覆盖掉nums1的数据,因此我们需要再开辟一个数组。
方法一:额外数组+双指针(从前遍历)
代码:
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int i = 0;
int j = 0;
int*res = (int*)malloc(sizeof(int)*(m+n));
int k = 0;
while(i<m||j<n)
{
if(i>=m)
{
res[k++] = nums2[j++];
continue;
}
if(j>=n)
{
res[k++] = nums1[i++];
continue;
}
if(nums1[i]<=nums2[j])
{
res[k++] = nums1[i++];
continue;
}
else
{
res[k++] = nums2[j++];
}
}
for(k=0;k<m+n;k++)
{
nums1[k] = res[k];
}
free(res);
}
方法二:原数组+双指针
我们可以直接从后遍历,这样就可以避免开辟新数组。
代码:
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int i = m-1;
int j = n-1;
int k = m+n-1;
while(i>=0||j>=0)
{
if(i<0)
{
nums1[k--] = nums2[j--];
continue;
}
if(j<0)
{
nums1[k--] = nums1[i--];
continue;
}
if(nums1[i]>=nums2[j])
{
nums1[k--] = nums1[i--];
continue;
}
if(nums1[i]<nums2[j])
{
nums1[k--] = nums2[j--];
continue;
}
}
}