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]

自己的算法以及代码

• 我的思路是：在合并的时候就直接进行比较。用了一个辅助数组C，用来暂时存放排序后的数组。
• 分了三种情况：
  • A、B 都未遍历完，比较大小，小的先放到数组 C 中；
  • A 未遍历完、B 完，将 剩余的A 全部存入 C 中；
  • B 未遍历完、A 完，将 剩余的B 全部存入 C 中；

public class Solution {

    public static void merge(int[] A, int m, int[] B, int n) {
        int i = 0;
        int j = 0;
        int k = 0;
        int len = m + n;
        int[] C = new int[len];
        while ((i < m) && (j < n)){
            if (A[i] <= B[j]){
                C[k++] = A[i++];
            }
            else{
                C[k++] = B[j++];
            }
        }
        while (i < m){
            C[k++] = A[i++];
        }
        while (j < n){
            C[k++] = B[j++];
        }

        for (int kk = 0; kk < C.length; kk++){
            A[kk] = C[kk];
        }
    }

    public static void main(String[] args) {
        int[] A = {1,2,3,0,0,0};
        int[] B = {2,5,6};
        int m = 3;
        int n = 3;
        merge(A, m, B, n);
    }
}