[路飞]_leetcode-面试题 10.01-合并排序的数组

706 阅读2分钟

「这是我参与2022首次更文挑战的第21天,活动详情查看:2022首次更文挑战

[题目地址] [B站地址]

给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。

初始化 A 和 B 的元素数量分别为 m 和 n

示例:

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6],       n = 3

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

说明:

  • A.length == n + m

解题思路

因为输入数组 A,B都是有序的数组,所以我们在合并的过程中只要保证将数组 B 中的元素插入合适的位置,就能保证合并后的数组依然是有序的。
这里我们定义两个指针 a,b,分别指向数组 A,B 的最后一个元素。
每次让 a 向前找到第一个小于等于 b 指向元素的位置(如果初始 a 指向元素小于等于 b 指向元素,则无需移动),然后将 a 指针后面的元素统一向后移动一位,在空出的位置插入 b 指向元素。这样就保证了每次插入 B 数组中的元素都在合适的位置,插入后,数组 A 依然有序。
向前移动 b 指针,重复上述过程,直到处理完数组 B 中的所有元素,就完成了将 B 合并入 A 的过程。

动画演示

leetcode-面试题 10.01-合并排序的数组.gif

代码实现

var merge = function (A, m, B, n) {
  // 初始化指针 a 指向 A 数组的末尾元素
  let a = m - 1,
    // 初始化指针 b 指向 B 数组的末尾元素
    b = n - 1
  // 从后向前的处理 B 数组中的每一个元素
  while (b >= 0) {
    // 让 a 指向第一个小于等于 b 指向元素的位置
    while (A[a] > B[b]) a--
    // 将 a 指针后面的元素统一向后移动一位
    move(A, a)
    // 将 b 指向元素插入空出的位置
    A[a + 1] = B[b]
    // 向前移动 b 指针处理剩余元素
    b--
  }
  /*
   * 传入数组和指定下标,将下标后面的元素统一向后移动一位
   * @param {number[]} arr
   * @param {number} ind
   * * @return {void}
   */
  function move(arr, ind) {
    // 从后向前将每一个元素向后移动一位,直到指定下标为止
    for (let i = arr.length - 2; i > ind; i--) {
      arr[i + 1] = arr[i]
    }
  }
}

至此我们就完成了 leetcode-面试题 10.01-合并排序的数组

如有任何问题或建议,欢迎留言讨论!👏🏻👏🏻👏🏻