前端算法二之leetcode实例88-合并两个有序数组

364 阅读1分钟

题目

给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。 你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。

示例:

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]

思路

  • 因为 nums1 的空间都集中在后面,所以从后向前处理排序的数据会更好,节省空间,一边遍历一边将值填充进去
  • 设置索引 len1 和 len2 分别指向 nums1 和 nums2 的有数字尾部,从尾部值开始比较遍历,同时设置指针 len 指向 合并后的nums1 的最末尾,每次遍历比较值大小之后,则进行填充。因为nums1和nums2都是有序整数数组,从后往前每项比较之后得出的最大值都要比nums1中排在前面的数值大
  • nums1和nums2的长度不确定,分两种情况:
  1. len1>=len2,nums2遍历比较完成后,nums1中还剩下len1-len2项,这些项还按照原来的排序即可,不需要做变动;
  2. len1<len2,nums1遍历比较完成后,nums2中还剩下len2-len1项,把这些剩下的项放在len1中头部,由于这些剩下的项已经是排过序的,所以无需再做比较。
  • 时间复杂度:O( n )

代码

var merge = function(nums1, m, nums2, n) {
    let len1 = m -1
    let len2 = n -1
    let len = m + n -1 
    while(len2 >= 0){
        if(len1<0){
            nums1[len--] = nums2[len2--]
            continue
        }
        nums1[len--] = nums1[len1]>nums2[len2]?nums1[len1--]:nums2[len2--]
    }
};

执行结果