gold-10-1


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

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

示例:

输入:
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

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        int i = m - 1;
        int j = A.length - 1;

        while (i >= 0) {
            A[j] = A[i];
            i--;
            j--;
        }

        int aIndex = ++j;
        int bIndex = 0;
        i = 0;

        while (aIndex < A.length || bIndex < n) {
            if (aIndex == A.length || (bIndex < n && A[aIndex] > B[bIndex])) {
                A[i] = B[bIndex];
                i++;
                bIndex++;
            } else {
                A[i] = A[aIndex];
                i++;
                aIndex++;
            }
        }
    }
}
//leetcode submit region end(Prohibit modification and deletion)

文章作者: 倪春恩
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 倪春恩 !