给定两个排序后的数组 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的后面,组成新的数组A再排序,时间复杂度主要花在排序上,最小也是O(m+n)log(m+n) (使用快排的情况)
从后向前数组遍历:因为 nums1 的空间都集中在后面,所以从后向前处理排序的数据会更好,节省空间,一边遍历一边将值填充进去
设置指针 len1 和 len2 分别指向 A 和 B 的有数字尾部,从尾部值开始比较遍历,同时设置指针 len 指向 A 的最末尾,每次遍历比较值大小之后,将大的值填充到指针len位置;
当 len1<0 时遍历结束,此时 B 中可能还有数据,将其直接拷贝到A的前面,最后得到结果数组A
class Solution {
public:
void merge(vector<int>& A, int m, vector<int>& B, int n) {
//从后面数组遍历,时间复杂度O(m+n)
int len1 = m-1;
int len2 = n-1;
int len = m+n-1;
while(len1>=0&&len2>=0){
if(A[len1]>B[len2]){
A[len] = A[len1];
len--;
len1--;
}
else{
A[len] = B[len2];
len--;
len2--;
}
}
while(len2>=0){
A[len2] = B[len2];
len2--;
}
}
};