【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--];
}
}
}