leetcode面试题 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

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