題目
合并兩個排序的整數(shù)數(shù)組A和B變成一個新的數(shù)組。
注意事項
你可以假設A具有足夠的空間(A數(shù)組的大小大于或等于m+n)去添加B中的元素雁佳。
樣例
給出 A = [1, 2, 3, empty, empty], B = [4, 5]
合并之后 A 將變成 [1,2,3,4,5]
代碼
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 i = m-1, j = n-1, index = m + n - 1;
while (i >= 0 && j >= 0) {
if (A[i] > B[j]) {
A[index--] = A[i--];
} else {
A[index--] = B[j--];
}
}
while (i >= 0) {
A[index--] = A[i--];
}
while (j >= 0) {
A[index--] = B[j--];
}
}
}