给定两个排序后的数组 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
解题思路:非常典型的两个有序表的合并,并且给出了将b并入a之中。能快速想到的方法就是对b中每个元素在a中找到他的插入位置。
但是在之前的学习过程中我们已经接触了sort这个排序函数,所以其实我们可以将a、b数组二合一,并为这一整个数组进行排序即可。
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());
}
};