力扣每日一题(2024.06.18):合并两个有序数组

112 阅读2分钟

两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。

注意: 最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。

示例:

输入: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解释: 需要合并 [1,2,3][2,5,6] 。
合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。

解析: 此处我想到了三种实现思路:

  1. 两个数组从小到大对比
  2. 两个数组从大到小对比(最优)
  3. 将第二个数组追加到第一个数组,然后采用冒泡算法排序

对于实现两个数组的合并,最优解最好是一遍过,让时间复杂度为O(m+n)O(m+n),即每次取出两个数组中的最大值放到第一个数组的尾部。此处采用第2种方式实现,代码实现如下:

public void merge(int[] nums1, int m, int[] nums2, int n) {
    int l=nums1.length-1;
    for(int i=1,j=1; j<=n;l--){
        // 第二个数组放到前面的条件:
        // 1.第一个数组没有数据 m=0
        // 2.当第二个数组取出的数比第一个数组所有数据都小时  m-i<0  此处i为第一个数组中数据移动的次数
        // 3.当第二个数组取出的数比第一个数组取出的数小时
        // 注意:此处2和3分开讨论的原因是如果第二个数组比第一个数组所有数字都小时,由于第一个数组已经移动了m次,所以第二个数组的数字
        // 直接插入即可,因为第一个数组再取数时会数组越界
        if(m==0 || m-i<0 || nums1[m-i]<=nums2[n-j]){
            nums1[l]=nums2[n-j];
            j++;
        }else{
            nums1[l]=nums1[m-i];
            i++;
        }
    }
}

欢迎各位指正和讨论!