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

题目

给定两个排序后的数组 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sorted-merge-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

这里只用原地修改A的方法。
由于确保A有足够的空间，那么最终A中元素的长度为 m+n，从这个索引位置p开始，利用双指针在A、B中从后往前遍历，较大者放入p的位置，p往前移动。

代码

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