给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
来源 https://leetcode-cn.com/problems/merge-sorted-array/submissions/
leetcode 88题
一开始我的思路是用多层循环完成比较和插入的过程。但到了实际运用中发现不仅代码很长而且实现较为困难。
第二种思路是两个数组直接合并,然后对合并后的数组进行排序。这个实现较为简单,但也需要较长的时间,所以我没有用上面两种方法。
我的思路是:因为两个数组都是有序的,所以对两个数组相同位置可以比较。而且容易想到从后往前比较是比较方便的(因为从前往后还要考虑到后面所有数字的移动)。
这个方法是这样的:从nums1的最大数字nums1[m-1]和nums2的最大数字nums[n-1]开始比较。较大的那个放到最后面nums1[n+m-1]处。然后小的那个数字不变(而大的数字所在的位置就不需要考虑了。如果从前往后比较,赋值的时候要考虑数字所在的位置!),这个数字和大的那个数组的前一位数字进行比较,再将大的那个数字赋值到nums2[n+m-2]处……一直进行到能比较的数字结束,即数字所在的位置大于等于0.
这样结束了吗?其实还没有。我们来考虑一下当nums1[0]和nums2[0]比较的过程。
如果nums1[0]>nums2[0]。 我们就在nums1[1]处赋值nums1[0],此时数字的位置已经在最前面了,所以就不再进行比较了,程序结束。但是此时的nums1[0]还是nums1[0] ,而不是更小的nums2[0]。 与答案不符。
那其实我们只要在最外面进行比较,用if将两个分别在(0,0)处的数字比较再赋值不就行了,或者直接用Array.Copy。
代码如下
public class Solution {
public void Merge(int[] nums1, int m, int[] nums2, int n) {
int i=m-1;
int j=n-1;
for(int k=m+n-1;k>=0;k--)
{
while(i>=0&&j>=0)
{
if(nums1[i]<nums2[j])
{
nums1[k]=nums2[j];
j--;
break;
}
else
{
nums1[k]=nums1[i];
i--;
break;
}
}
}
Array.Copy(nums2,nums1,j+1);;
}
}
这个代码中主要注意的是两个 break 的使用,如果没有break,他在while中会一直进行下去,最终只完成最后(时间层面上的最后)一个数字的移动。
应该有其他更简单的代码可以实现相同的内容,例如直接用三元运算符这类的,或者用nums1[–i]这类的方式直接简化。
实现速度较快,时间复杂度为O(m+n)。