给定两个排序后的数组 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]
可以另外开辟空间通过比较存入,但是也可以原地操作
从后往前
i指向m-1(A内容的末尾),j指向A末尾,j指向B末尾
通过比较哪个大哪个放在末尾
需要注意下A有可能m=0,所以判断当i<0时只将B的内容放入A后面
class Solution { public: void merge(vector<int>& A, int m, vector<int>& B, int n) { int i=m-1,j=m+n-1,k=n-1; while(k>=0){//从后往前遍历,哪个大哪个追加到末尾 if(i<0||A[i]<=B[k]){//i<0是避免A为[0,0]且m=0的情况(易错点) A[j--]=B[k--]; } else{ A[j--]=A[i--]; } } } };