使用 JavaScript 解答 LeetCode 第 88 题,合并两个有序数组,简单难度。

题目

原题链接

本题目是 LeetCode 题库第 88 题,由此访问。

题目描述

给你两个有序整数数组 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]

解题

AC 样例

/**
 * @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 i1 = m - 1, // nums1指针
        i2 = n - 1, // nums2指针
        tail = m + n - 1; // 合并数组尾指针

    while(i2 >= 0){
        if (nums1[i1] > nums2[i2]) { 
            nums1[tail] = nums1[i1];
            i1--;
            tail--;            
        } else {
            nums1[tail] = nums2[i2];
            i2--;
            tail--;
        }
    }
};

解题思路

类似于归并排序算法,将两个已排序数组合并为一个排序数组,分别从两个数组的相同一端起步进行数据检索,并比较其大小。一般可以选择从数组头部或尾部开始检索,考虑到题目中所提供测试用例是自小到大的排序顺序,而从头部开始检索时,有大概率对后续的所有数据进行平移操作,时间复杂度较大。从数组后端开始检索可以避免额外的时间开销。

自数组尾部开始检索数据,如其一数据大于另一数据,则其较大值写入合并数组尾指针位置,而后尾指针同较大数据所在数组指针一并左移,进行下一轮比较。按照题目要求,数组 nums1 规模至少为 m+n,远大于 nums2 数组,故当 nums2 的指针到达数组头部时,上述检索和比较值的过程循环结束。