合并两个排序的整数数组A和B变成一个新的数组。
注意事项
你可以假设A具有足够的空间(A数组的大小大于或等于m+n)去添加B中的元素。
给出 A = [1, 2, 3, empty, empty]
, B = [4, 5]
合并之后 A 将变成 [1,2,3,4,5]
class Solution {
public:
/*
* @param A: sorted integer array A which has m elements, but size of A is m+n
* @param m: An integer
* @param B: sorted integer array B which has n elements
* @param n: An integer
* @return: nothing
*/
void mergeSortedArray(int A[], int m, int B[], int n) {
// write your code here
if(n==0){//B数组为空
return ;
}
vector<int> mn;
int k=0,mb=0;
for(int i=0;i<m;i++){
if(A[i]<=B[k]){
mn.push_back(A[i]);
mb++;
if(mb==m){//如果A数组所有元素都添加到mb中退出循环
break;
}
continue;
}
if(A[i]>B[k]){
mn.push_back(B[k]);
i--;
k++;
if(k==n){//如果B数组所有元素都添加到mb中退出循环
break;
}
}
}
for(int i=mb;i<m;i++){//如果A数组中还有剩余元素将其添加到mb中
mn.push_back(A[i]);
}
for(int i=k;i<n;i++){//若果B数组中还有剩余的元素将其添加到mb中
mn.push_back(B[i]);
}
for(int i=0;i<m+n;i++){//将mb数组中的元素赋给A数组
A[i]=mn[i];
}
return ;
}
};