每日一题——合并两个有序的数组

题目

---

给出一个有序的整数数组 A 和有序的整数数组 B ，请将数组 B 合并到数组 A 中，变成一个有序的升序数组

数据范围：0≤n,m≤100，∣A_i∣<=100，∣B_i∣<=100

注意：

1.保证 A 数组有足够的空间存放 B 数组的元素， A 和 B 中初始的元素数目分别为 m 和 n，A的数组空间大小为 m+n。

2.不要返回合并的数组，将数组 B 的数据合并到 A 里面就好了，且后台会自动将合并后的数组 A 的内容打印出来，所以也不需要自己打印。

3. A 数组在[0,m-1]的范围也是有序的。

示例1

输入：
[4,5,6],[1,2,3]
返回值：
[1,2,3,4,5,6]
说明：
A数组为[4,5,6]，B数组为[1,2,3]，后台程序会预先将A扩容为[4,5,6,0,0,0]，B还是为[1,2,3]，m=3，n=3，传入到函数merge里面，然后请同学完成merge函数，将B的数据合并A里面，最后后台程序输出A数组

示例2

输入：
[1,2,3],[2,5,6]
返回值：
[1,2,2,3,5,6]

思路

---

数组从后往前比较，较大的放入A的尾部。如果B先遍历完，A剩下的已经在数组前面符合题意的位置上。如果A先遍历完，则将B剩下的元素依次放入A中。

解答代码

---

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int i = m-1;
        int j = n-1;
        int k = m+n-1;
        while (i>=0 && j>=0) {
            // 从后往前比较，将较大的元素放到A的尾部
            A[k--] = A[i]<=B[j] ? B[j--] : A[i--];
        }
        // 如果B中还有元素，则按顺序放到A里
        while (j>=0) {
            A[k--] = B[j--];
        }
    }
};