一、题目
给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。
初始化 A 和 B 的元素数量分别为 m 和 n。
示例:
输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
说明:A.length == n + m
二、实现
从后向前比较,每次取数组A和B中较大的元素,放到数组A的后面
class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int pa = m-1;
int pb = n-1;
int cur = m+n-1;
while(pa>=0 || pb>=0){
if(pa<0){
A[cur--] = B[pb--];
continue;
}
if(pb<0){
A[cur--] = A[pa--];
continue;
}
if(A[pa]>B[pb]){
A[cur--] = A[pa--];
}else{
A[cur--] = B[pb--];
}
}
}
}