合并两个排序的整数数组A和B变成一个新的数组。
样例
给出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
最简单的思路,先把两个数组的数据放入一个数组中,然后再排序就可以了,但是这样做时间复杂度还是挺高的,原因就在于人家本来就是已经排好数的数组了,所以更简单的方法是利用双指针(这里用迭代器),总是把指针指向小的那个元素放入新的数组,这样只需要两个数组都遍历一次加上一些简单的比较就好了。代码也简单明了,有些需要注意的地方写在下面。
利用while循环控制双指针:
这个有两个原因,一是++的条件我们是在if里的,二是循环变量也应该定义在循环外(终止循环后我们还要用到这些变量),这样的话用for循环就没有意义了。
只要有一个指针到头我们就应该跳出循环:
因为这样的话这个数组的数肯定都是小于另外一个数组指针之后的数的,我们只需要拿过来一个一个放入即可。就不需要再进行比较了,这样做也能节省时间。
code:
vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
vector<int> res;
auto beg_a=A.begin();
auto beg_b=B.begin();
while(beg_a!=A.end()&&beg_b!=B.end()) //双指针,只要有一个到头就停止。
{
if(*beg_a<*beg_b)
{
res.push_back(*beg_a);
beg_a++;
}
if(*beg_a>*beg_b)
{
res.push_back(*beg_b);
beg_b++;
}
if(*beg_a==*beg_b)
{
res.push_back(*beg_a);
beg_a++;
res.push_back(*beg_b);
beg_b++;
}
}
while(beg_a!=A.end()) //看哪一个没有遍历完,把后续都添加进去
{
res.push_back(*beg_a);
beg_a++;
}
while(beg_b!=B.end())
{
res.push_back(*beg_b);
beg_b++;
}
return res;
// write your code here
}
over
网友评论