Leetcode 面试题 10.01. 合并排序的数组 (每日一题 20210616)

给定两个排序后的数组 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