LintCode算法之合并排序数组 II

合并两个有序升序的整数数组A和B变成一个新的数组。新数组也要有序。
样例
输入: A=[1,2,3,4], B=[2,4,5,6]
输出: [1,2,2,3,4,4,5,6]

 public int[] mergeSortedArray(int[] A, int[] B) {
        int arr[]=new int[A.length+B.length];
        int p1=0;
        int p2=0;
        int i=0;
        //依次取A数组和B数组的最小值，直到某个数组取空了
        while(p1<A.length&&p2<B.length){
             if(A[p1] <= B[p2]){
                 arr[i++] = A[p1++]; 
             }else {
                 arr[i++] =B[p2++]; 
             }
        }
         //如果A数组还没有遍历完，则剩下的数均比B数组中的大，写入arr最后即可
         while(p1 < A.length){
        	 arr[i++] = A[p1++];
         }  
         //如果B数组还没有遍历完，则剩下的数均比A数组中的大，写入arr最后即可
         while(p2 < B.length) {
         	arr[i++] = B[p2++];   
         }
         return arr;
    }