面试题 10.01. 合并排序的数组

Posted 2021-02-28 hunter-w

描述

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

思路

• 方法1 直接排序
  将B元素添加到A中，对A进行排序即可。

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        for(int i=0;i<n;i++)
            A[m+i]=B[i];
        sort(A.begin(),A.end());
    }
};

• 方法2 逆序双指针原地排序
  考虑到A的后面有足够的容量包含B元素，这样使用双指针比较AB中元素，较大者放在A后面。

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        int i=m-1,j=n-1;
        int tail=m+n-1;
        int cur=0;
        while(i>=0||j>=0){
            if(i==-1)
                cur=B[j--];
            else if(j==-1)
                cur=A[i--];
            else if(A[i]>B[j])
                cur=A[i--];
            else
                cur=B[j--];
            A[tail--]=cur;
        }
    }
};