力扣 面试题 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]

方法1：正向双指针

需要额外空间

Java实现

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        int l = 0, r = 0;
        List<Integer> res = new ArrayList<>();

        while (l < m && r < n) {
            if (A[l] <= B[r]) res.add(A[l++]);
            else res.add(B[r++]);
        }

        while (l < m) res.add(A[l++]);
        while (r < n) res.add(B[r++]);

        for (int i = 0; i < m + n; i++) {
            A[i] = res.get(i);
        }
    }
}

方法2：逆向双指针

Java实现

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        int l = m - 1, r = n - 1;
        int flag = m + n - 1;

        while (l >= 0 && r >= 0) {
            if (B[r] >= A[l]) A[flag--] = B[r--];
            else A[flag--] = A[l--];
        }
        while (l >= 0) A[flag--] = A[l--];
        while (r >= 0) A[flag--] = B[r--];
    }
}