python 合并排序的数组

｜ 合并排序的数组

给定两个排序后的数组 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:
    def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
        """
        Do not return anything, modify A in-place instead.
        解题思路: 
        1.把数组B 追加到数组A的第m个元素的后面
        2.利用快速排序对数组A进行排序
        """
        
        A[m:] = B
        self.quick_sort(A, 0, m+n-1)

    
    def quick_sort(self, A, first, last):
        # 初始化p, q指针 指向首、尾元素  
        if first >= last:
            return
        p = first
        q = last

        while q > p:
            if A[q] >= A[first]:
                q -= 1
            elif A[p] < A[first]:
                p += 1
            else:
                A[p], A[q] = A[q], A[p]
        A[p] = A[first]

        # 对分割左侧的元素排序
        self.quick_sort(A, first, q-1)
        self.quick_sort(A, q+1, last)

# 题解二
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.
        解题思路: 逆向双指针(比较不好理解)，下面代码有详细注释
        """
        # 如果m==0 则直接返回B
        if m == 0:
            A[m:] = B

        while m > 0 and n > 0:
            # 当B数组元素大时 把当前元素之间放置A数组后面
            if B[n-1] > A[m-1]:
                A[m+n-1] = B[n-1]
                n -= 1 # 向前移动一个单位
            else:
                # 当A数组元素比较大的时候，把当前元素调到后面 
                A[m+n-1] = A[m-1]  
                m -= 1  # 向后移动一个单位
        
        # 特殊情况处理 如果A数组元素已经对比结束，退出循环时，B数组中还有未对比的元素，则直接赋值给A元素，以索引进行切割
        if m == 0: 
            A[:n] = B[:n]