3.3 面试题 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和B共有元素n个，那么目的就是有序的填慢A中的前n个空。
• 用end表示A中要填满区间的尾index。endA表示A中元素的尾index, endB表示B的尾index。
• 同时从尾部遍历A和B。end > -1为循环终止条件。
• 当endA等于-1时，说明A中的元素遍历完了。若B中还有元素，填充到A中。
• 当endB等于-1时，说明B中的已经填完了，此时直接break。
• 当endA和endB都不为-1时，选择大的从尾部填充。

代码

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        int end = m + n - 1;
        int endA = m - 1;
        int endB = n - 1;

        while ( end > -1 ) {
            if ( endA == -1 ) 
                A[end--] = B[endB--];
            else if ( endB == -1 )
                break;
            else {
                if ( A[endA] >= B[endB] )
                    A[end--] = A[endA--];
                else 
                    A[end--] = B[endB--];
            }
        }

        return;
    }
};