题目描述
合并两个排序的整数数组A和B变成一个新的数组。
样例给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
挑战
你能否优化你的算法,如果其中一个数组很大而另一个数组很小?
分析
对于合并两个已经排好序的数组问题,直接使用归并排序的思想。当两个数组比较的时候,选择较小的那个数,并将其压入返回的数组里面,然后移动到下一个位置。当一个数组被比较完时,直接将另一个数组全部压入返回数组即可。最终的时候复杂度O(n),空间复杂度O(n)。
程序
class Solution {
public:
/**
* @param A: sorted integer array A
* @param B: sorted integer array B
* @return: A new sorted integer array
*/
vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
// write your code here
vector<int> C;
int size_A = A.size();
int size_B = B.size();
int i = 0, j = 0;
while(i < size_A && j < size_B){
if(A[i]<B[j]){
C.push_back(A[i++]);
}
else{
C.push_back(B[j++]);
}
}
while(i < size_A){
C.push_back(A[i++]);
}
while(j < size_B){
C.push_back(B[j++]);
}
return C;
}
};