Java 合並兩個排序數組

 題目:將兩個排序好的數組組成一個新的排序好的數組,給出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;
    }
}