题目描述:
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n
方法1. 创建一个新数组对两个数组排列
merge函数中给的nums1Size和nums2Size感觉没有用,而且会误导你,nums1数组的长度是m而不是nums1Size;nums2数组的长度同样是n.
设置一个数组和三个变量。三个变量分别记录当前nums1数组元素位置、nums2数组元素位置和新建数组的当前位置。依次比较两个数组,选择小的放入新建数组中,并向后移动位置,直到遍历结束。结束后还要判断哪个数组有剩余元素,依次存放新数组中。
最后遍历新数组,依次将元素放入nums1中。该方法是本人自己写的,空间复杂度为O(N)。
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int allnumsSize = m+n;
int mer[allnumsSize];
int temp = 0;
int flag1 = 0, flag2 = 0;
while((flag1<m)&&(flag2<n))
{
if(nums1[flag1]<=nums2[flag2])
{
mer[temp] = nums1[flag1];
++flag1;
++temp;
}
else
{
mer[temp] = nums2[flag2];
++flag2;
++temp;
}
}
while(flag1<m)
{
mer[temp] = nums1[flag1];
++flag1;
++temp;
}
while(flag2<n)
{
mer[temp] = nums2[flag2];
++flag2;
++temp;
}
for(int i=0; i<allnumsSize; ++i)
{
nums1[i] = mer[i];
}
}
上面程序可以简化,但要注意while中的判断首先要判断其中一个数组是否已经被全部插入新数组,否则会继续判断两个数组中的元素大小,报错。
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int allnumsSize = m+n;
int* mer = (int*)malloc(sizeof(int)*allnumsSize);
int temp = 0;
int flag1 = 0, flag2 = 0;
while(flag1<m || flag2<n)
{
if(flag1==m)
{
mer[temp] = nums2[flag2];
++flag2;
}
else if(flag2==n)
{
mer[temp] = nums1[flag1];
++flag1;
}
else if(nums1[flag1]<nums2[flag2])
{
mer[temp] = nums1[flag1];
++flag1;
}
else if(nums1[flag1]>=nums2[flag2])
{
mer[temp] = nums2[flag2];
++flag2;
}
++temp;
}
for(int i=0; i<allnumsSize; ++i)
{
nums1[i] = mer[i];
}
}
方法2. 从后依次遍历两个数组
该方法不需要创建数组,空间复杂度为O(1)
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n){
int allnumsSize = m+n;
int flag1 = m-1;
int flag2 = n-1;
int temp = allnumsSize-1;
while(flag1>=0 || flag2>=0)
{
if(flag1<0)
{
nums1[temp] = nums2[flag2];
flag2--;
}
else if(flag2<0)
{
nums1[temp] = nums1[flag1];
flag1--;
}
else if(nums2[flag2]>nums1[flag1])
{
nums1[temp] = nums2[flag2];
flag2--;
}
else if(nums2[flag2]<=nums1[flag1])
{
nums1[temp] = nums1[flag1];
flag1--;
}
temp--;
}
}
官方给出的方法是使用指针,个人感觉没有设置变量容易理解。