合并排序数组 II——LintCode

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

您在真实的面试中是否遇到过这个题？ 

Yes

样例

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

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

注意

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

//m是A数组中不为空的元素，n是B数组中的元素

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) {
        // write your code here
        for(int i=0;i<n;i++)
			A[m+i] = B[i];
		Arrays.sort(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) {
        // write your code here
        int pa = m - 1, pb = n - 1, p = m + n - 1;
		while (pa >= 0 && pb >= 0) {
			if (A[pa] >= B[pb])
				A[p--] = A[pa--];
			else
				A[p--] = B[pb--];
		}
		while(pb >= 0)
			A[p--] = B[pb--];
    }
}