给定两个排序后的数组 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
class Solution {
public:
void merge(vector<int>& A, int m, vector<int>& B, int n) {
int i = m+n-1;
int j = m-1;
int k = n-1;
while(i>=0){
if(j>=0&&k>=0){
if(A[j]>B[k]){
A[i] = A[j];
j--;
}
else{
A[i] =B[k];
k--;
}
}else if(j>= 0){
A[i] = A[j];
j--;
}else if(k>=0)
{
A[i] = B[k];
k--;
}
i--;
}
}
};
python3
class Solution:
def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
idx1 = m - 1
idx2 = n - 1
cur = m + n - 1 # 添加 cur 指针追踪位置
while idx1 > -1 and idx2 > -1:
# print(A)
if A[idx1] < B[idx2]:
A[cur] = B[idx2]
cur -= 1
idx2 -= 1
else:
A[cur] = A[idx1]
cur -= 1
idx1 -= 1
if idx2 != -1: A[:idx2 + 1] = B[:idx2 + 1] # 比较完B还有剩下的,全填到A前面即可
return A