题目描述
给你两个按非递减顺序排列的整数数组
nums1
和nums2
,另有两个整数m
和n
,分别表示nums1
和nums2
中的元素数目。请你合并nums2
到nums1
中,使合并后的数组同样按非递减顺序排列。注意:最终,合并后数组不应由函数返回,而是存储在数组
nums1
中。为了应对这种情况,nums1
的初始长度为m+n
,其中前m
个元素表示应合并的元素,后n
个元素为0
,应忽略。nums2
的长度为n
。
思路:比较nums1和nums2两个数组最后一个元素(假设下标分别为i1和i2)哪一个大(比如nums1最后一个元素更大),把较大值放到nums1的最后一个位置。然后继续比较nums1的倒数第二个元素和num2的倒数第一个元素,把较大值放到nums1的倒数第二个元素位置处,以此类推。显而易见,这里是一个循环,那么循环的条件是什么呢?如果i1<0或者i2<0,那说明nums1和nums2中有一个数组被遍历完了,此时我们不应该继续循环,否则可能产生越界。
那我们继续思考一下,如果是因为i2<0而退出循环,那么就说明nums2中的所有元素都被挪到nums1中了,而nums1中的元素肯定也都在nums1中,这就表明所有的元素都已经被放到nums1中了,那么大功告成;然而,如果是因为i1<0而退出循环,那此时i2必定大于等于0,即i2中的元素还没有被挪完就退出循环了,那么在退出循环后必须把nums2中剩下的元素都挪到nums1中。
因此,我的实现代码如下:
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n)
{
int i1=m-1;
int i2=n-1;
int j=m+n-1;
while(i1>=0 && i2>=0)
{
if(nums1[i1] > nums2[i2])
{
nums1[j]=nums1[i1];
i1--;
j--;
}
else
{
nums1[j]=nums2[i2];
i2--;
j--;
}
}
//判断退出循环后是否是因为i1<0
while(i2>=0)
{
nums1[j]=nums2[i2];
i2--;
j--;
}
}