合并两个排序的整数数组A和B变成一个新的数组。
注意事项
你可以假设A具有足够的空间(A数组的大小大于或等于m+n)去添加B中的元素。
样例
给出 A = [1, 2, 3, empty, empty], B = [4, 5]
合并之后 A 将变成 [1,2,3,4,5]
最容易想到的方法是将B中的元素先添加到A中然后对A进行一次排序
从另外的角度上来看,这里与归并排序的原理是一致的,问题在于我们没有额外的数组空间去容纳新的序列,解决的办法是将A作为额外的数组空间,从尾部开始添加元素,利用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
// for(int i = 0;i < n; i++){
// A[m + n - i - 1] = B[n - i - 1];
// }
// Arrays.sort(A);
//从尾部开始填充
int k = m + n -1;
m--;
n--;
while(k>=0){
if(n<0 || (m>=0 && A[m] >= B[n])){
A[k--] = A[m--];
}else{
A[k--] = B[n--];
}
}
}
}