【LeetCode打卡】面试题10.01.合并排序的数组

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

说明:

A.length == n + m

代码示例

1、直接求解

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
       for(int i = m;i < m+n;i++){
            A[i] = B[i-m];
        }
        Arrays.sort(A);
    }
}

2、双指针

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        int[] temp = new int[m+n];
        int temp1 = 0;
        int a = 0,b = 0;
        while(a < m || b < n){
            if(a == m){
                temp1 = B[b++];
            }else if(b == n){
                temp1 = A[a++];
            }else if(A[a] <= B[b]){
                temp1 = A[a++];
            }else{
                temp1 = B[b++];
            }
            temp[a+b-1] = temp1;
        }
        for(int i = 0;i < m+n;i++){
            A[i] = temp[i];
        }
    }
}

3、逆向双指针

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
        int length = m+n-1;
        int temp1 = 0;
        int a = m-1,b = n-1;
        while(a >= 0 || b >= 0){
            if(a == -1){
                temp1 = B[b--];
            }else if(b == -1){
                temp1 = A[a--];
            }else if(A[a] >= B[b]){
                temp1 = A[a--];
            }else{
                temp1 = B[b--];
            }
            A[length--] = temp1;
        }
    }
}