合并两个有序数组

  • Post author:
  • Post category:其他


真题描述:给你两个有序整数数组 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 的末尾往前面填补:

如果提前遍历完的是 nums1 的有效部分,剩下的是 nums2。那么这时意味着 nums1 的头部空出来了,直接把 nums2 整个补到 nums1 前面去即可。

如果提前遍历完的是 nums2,剩下的是 nums1。由于容器本身就是 nums1,所以此时不必做任何额外的操作。

编码实现

/**
 * @param {number[]} nums1
 * @param {number} m
 * @param {number[]} nums2
 * @param {number} n
 * @return {void} Do not return anything, modify nums1 in-place instead.
 */
var merge = function(nums1, m, nums2, n) {
    let i = m - 1,j = n - 1, k = m + n - 1;
    while(i>= 0 && j>=0){
        if(nums1[i]>= nums2[j]){
            nums1[k] = nums1[i];
            k--;
            i--;
        }else{
            nums1[k] = nums2[j];
            k--;
            j--;
        }
    }

    while(j>=0){
        nums1[k] = nums2[j]
        k--;
        j--;
    }
    return  nums1
};



版权声明:本文为weixin_43939659原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。