问题描述:合并两个排序的整数数组A和B变成一个新的数组。
样例:给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
分析:首先要有两个变量i、j来遍历两个数组,只要A、B两个数组有一个没有遍历完,就要进行操作。考虑到A和B的长度不同,当A遍历完时,将B中剩余元素全部放入C尾;当B遍历完时,将A中剩余元素全部放入C尾。否则,A、B均未遍历完,比较A[i]与B[j]的大小,小的入C尾,然后让其对应的变量加1,直至两个数组全部遍历完。
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
// 定义一个int型数组C
vector<int> C;
int i = 0,j = 0;
// C++ 求数组的长度 A.size()
int aLen = A.size();
int bLen = B.size();
while(i != aLen || j != bLen){
if(i == aLen){
// 将B[j]加入到数组C的尾部
C.push_back(B[j]);
j++;
}else if(j == bLen){
C.push_back(A[i]);
i++;
}else{
if(A[i]<=B[j]){
C.push_back(A[i]);
i++;
}else{
C.push_back(B[j]);
j++;
}
}
}
return C;
}
};
如有错误,欢迎指出。