Java 合并两个排序数组

Posted 2020-10-04 李乐

 题目:将两个排序好的数组组成一个新的排序好的数组,给出A=[1,2,3,4]，B=[2,4,5,6]，返回 [1,2,2,3,4,4,5,6]

 

挑战

你能否优化你的算法，如果其中一个数组很大而另一个数组很小？

 

 

 思路: 两根指针分别指向两个数组的开头,每次取值两个指针位置较小的数，指针后移,直到其中一个数组结束,将另外一个数组指针后面的元素连接到新数组的后面即可

 关于挑战的情况 如果A[0]>B[B.length-1]||A[A.length-1]<B[0] ，可根据情况 直接将A连接到B的后面(或者B连接到A的后面)

代码如下

 

 public int[] mergeSortedArray(int[] A, int[] B) {
        int i=0;
        int j=0;
        int res[] = new int[A.length+B.length];
        int k=0;
        while(i<A.length&&j<B.length){
            if(A[i]<=B[j]){
                res[k++]=A[i++];
            }else{
                res[k++]=B[j++];
            }
        }

        if(i!=A.length){
            for(int a=i;a<A.length;a++){
                res[k++]=A[a];
            }
        }else{
            for(int b=j;b<B.length;b++){
                res[k++]=B[b];
            }

        }
        return res;
    }
}