LintCode之合并排序数组

题目描述：

我的代码：

 public class Solution {
     /*
      * @param A: sorted integer array A
      * @param B: sorted integer array B
      * @return: A new sorted integer array
      */
     public int[] mergeSortedArray(int[] A, int[] B) {
         // write your code here
         int[] a = new int[A.length+B.length];
         int p=0,q=0,count=0;
         //当有一个数组全都重新排好序之后就退出循环
         while(p<A.length && q<B.length) {
             if(A[p] <= B[q]) {
                 a[count++] = A[p];
                 p++;
             }else {
                 a[count++] = B[q];
                 q++;
             }
         }
         if(p < A.length) {
             for(int i=p; i<A.length; i++) {
                 a[count++] = A[i];
             }
         }
         if(q < B.length) {
             for(int i=q; i<B.length; i++) {
                 a[count++] = B[i];
             }
         }
         return a;
     }
 }