给定两个排序后的数组 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
链接:https://leetcode-cn.com/problems/sorted-merge-lcci
class Solution:
def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
"""
Do not return anything, modify A in-place instead.
"""
index_1, index_2, index_all = m - 1, n - 1, m + n -1
while index_1 >= 0 and index_2 >= 0:
if A[index_1] < B[index_2]:
A[index_all] = B[index_2]
index_all -= 1
index_2 -= 1
else:
A[index_all] = A[index_1]
index_all -= 1
index_1 -= 1
if index_2 >= 0:
A[0:index_2+1] = B[0:index_2+1]
return A