每日算法-10.01. 合并排序的数组

10.01. 合并排序的数组

面试题 10.01. 合并排序的数组
给定两个排序后的数组 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拼接到一起

class Solution:
    def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
        A[:]=A[:m]+B
        A.sort()

方法二

双指针
出处
https://leetcode.cn/problems/sorted-merge-lcci/solution/tu-jie-shuang-zhi-zhen-yuan-di-xiu-gai-by-z1m/

class Solution:
    def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
        a_end = -n-1
        b_end = -1
        i =1
        if m == 0:
            A[:] =B 
            return A
        if n== 0:
            return A 
        while i< n+m:
            if B[b_end] >= A[a_end]:
                A[-i] = B[b_end]
                b_end -= 1
            else:
                A[-i] = A[a_end]
                A[a_end]=0
                a_end -= 1

            if b_end == (-n-1):
                break
            if a_end == (-m-n-1):
                if b_end==-1:
                    A[-m - n:-i] = B[-n:]
                else:
                    A[-m - n:-i ] = B[-n:b_end+1]
                break
            i += 1