LeetCode 算法：合并排序的数组

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第 5 天，点击查看活动详情

合并排序的数组

原题地址

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

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

示例1:

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

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

示例2:

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

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

说明:

• A.length == n + m

思路分析

方法一

1. 看到题目的第一反应就是将两个数组组装起来，然后再进行排序；
2. 那么如何将数组组装起来呢，因为 A 的长度是 m，B 的长度是 n，那么可以使用 splice 的 API 将数组拼接起来；
3. 最开始使用的 sort 方法，但是由于 sort 方法并不能改变数组本身，因此不起作用；
4. 那么只能使用二次遍历进行排序，最后改变 A 的值。

方法二

1. 将 A 和 B 从后往前遍历，双指针移动；
2. 若 A[lenA] 为 undefined 并且 B 对应的元素大于 A 对应的元素，那么将 B 对应的元素赋值给 A[i]，否则将 A 对应的元素赋值给 A[i] 即可得到想要的结果。

AC 代码

方法一

/**
 * @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) {
    A.splice(m, n, ...B)
    for(let i = 0; i < A.length; i++) {
        for(let j = i + 1; j < A.length; j++) {
            if(A[i] > A[j]) {
                [A[i], A[j]] = [A[j], A[i]]
            }
        }
    }
};

结果：

• 执行结果： 通过
• 执行用时：60 ms, 在所有 JavaScript 提交中击败了75.22%的用户
• 内存消耗：42.7 MB, 在所有 JavaScript 提交中击败了5.08%的用户
• 通过测试用例：59 / 59

方法二

/**
 * @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) {
    const len = m + n
    let lenA = m - 1
    let lenB = n - 1
    for(let i = len - 1; i >= 0 && lenB >= 0; i--) {
        if(A[lenA] === undefined || B[lenB] >= A[lenA]) {
            A[i] = B[lenB]
            lenB--
        } else {
            A[i] = A[lenA]
            lenA--
        }
    }
};

结果：

• 执行结果： 通过
• 执行用时：64 ms, 在所有 JavaScript 提交中击败了49.85%的用户
• 内存消耗：41.1 MB, 在所有 JavaScript 提交中击败了63.88%的用户
• 通过测试用例：59 / 59

END