算法题解:合并两个有序数组-JavaScript

130 阅读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 中的元素。

输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
解释:需要合并 [1] 和 [] 。
合并结果是 [1] 。

输入:nums1 = [0], m = 0, nums2 = [1], n = 1
输出:[1]
解释:需要合并的数组是 [] 和 [1] 。
合并结果是 [1] 。
注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。

题解一:

/*
 * @lc app=leetcode.cn id=88 lang=javascript
 *
 * [88] 合并两个有序数组
 */

// @lc code=start
/**
 * @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) {
  // 使用splice组合两个数组后, 再进行排序
  nums1.splice(m, n, ...nums2);
  nums1.sort((a,b) => a - b);
  }
};
// @lc code=end

题解二: 使用两个指针进行比较

Untitled.png

/*
 * @lc app=leetcode.cn id=88 lang=javascript
 *
 * [88] 合并两个有序数组
 */

// @lc code=start
/**
 * @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 p1 = 0,
    p2 = 0;
  // 新建个临时变量
  const sortedArray = new Array(m + n).fill(0);

  let cur;
  // 在遍历过程中排序,当迭代完成两个数组时循环结束
  while (p1 < m || p2 < n) {
    if (p1 === m) {
			// 如果下标p1 与 长度m相等, 表明num1数组已经迭代结束,则需要迭代num2, 下面同理
      cur = nums2[p2++];
    } else if (p2 === n) {
      cur = nums1[p1++];
    } else if (nums1[p1] < nums2[p2]) {
      cur = nums1[p1++];
    } else {
      cur = nums2[p2++];
    }

    // p1 和 p2 指针始终递增, -1是获取正确的下标
    sortedArray[p1 + p2 - 1] = cur;
  }

  for (let i = 0; i !== m + n; i++) {
    nums1[i] = sortedArray[i];
  }
};
// @lc code=end

题解三: 使用三指针尾插

/*
 * @lc app=leetcode.cn id=88 lang=javascript
 *
 * [88] 合并两个有序数组
 */

// @lc code=start
/**
 * @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,
    tail = m + n - 1,
    cur;

  while (i >= 0 || j >= 0) {
    if (i < 0) {
      cur = nums2[j--];
    } else if (j < 0) {
      cur = nums1[i--];
    } else if (nums1[i] < nums2[j]) {
      cur = nums2[j--];
    } else {
      cur = nums1[i--];
    }
    nums1[tail--] = cur;
  }
};
// @lc code=end

c.png