【LintCode】合并排序数组 II

合并两个排序的整数数组A和B变成一个新的数组。

 注意事项

你可以假设A具有足够的空间（A数组的大小大于或等于m+n）去添加B中的元素

样例

给出 A = [1, 2, 3, empty, empty], B = [4, 5]

合并之后 A 将变成 [1,2,3,4,5]

思路：也是道归并排序题。不过排序可以不增加辅助数组，直接从A的末端开始修改元素

代码如下：

class Solution {
    /**
     * @param A: sorted integer array A which has m elements, 
     *           but size of A is m+n
     * @param B: sorted integer array B which has n elements
     * @return: void
     */
    public void mergeSortedArray(int[] A, int m, int[] B, int n) {
        int ilen = m + n -1;
        int ia = m -1;
        int ib = n -1;
        
        while(ia >= 0 && ib>= 0 && ilen >= 0)
        {
            if(A[ia] > B[ib])
            {
                A[ilen--] = A[ia--];
            }
            else
            {
                A[ilen--] = B[ib--];
            }
        }
        
        while(ia >= 0 && ilen >= 0)
        {
            A[ilen--] = A[ia--];
        }
        
        while(ib >= 0 && ilen >= 0)
        {
            A[ilen--] = B[ib--];
        }
    }
}