合并兩個(gè)排序的整數(shù)數(shù)組A和B變成一個(gè)新的數(shù)組。
樣例
給出A=[1,2,3,4],B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]
最簡單的思路,先把兩個(gè)數(shù)組的數(shù)據(jù)放入一個(gè)數(shù)組中,然后再排序就可以了,但是這樣做時(shí)間復(fù)雜度還是挺高的,原因就在于人家本來就是已經(jīng)排好數(shù)的數(shù)組了,所以更簡單的方法是利用雙指針(這里用迭代器),總是把指針指向小的那個(gè)元素放入新的數(shù)組,這樣只需要兩個(gè)數(shù)組都遍歷一次加上一些簡單的比較就好了。代碼也簡單明了,有些需要注意的地方寫在下面。
利用while循環(huán)控制雙指針:
這個(gè)有兩個(gè)原因,一是++的條件我們是在if里的,二是循環(huán)變量也應(yīng)該定義在循環(huán)外(終止循環(huán)后我們還要用到這些變量),這樣的話用for循環(huán)就沒有意義了。
只要有一個(gè)指針到頭我們就應(yīng)該跳出循環(huán):
因?yàn)檫@樣的話這個(gè)數(shù)組的數(shù)肯定都是小于另外一個(gè)數(shù)組指針之后的數(shù)的,我們只需要拿過來一個(gè)一個(gè)放入即可。就不需要再進(jìn)行比較了,這樣做也能節(jié)省時(shí)間。
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()) //雙指針,只要有一個(gè)到頭就停止。
{
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()) //看哪一個(gè)沒有遍歷完,把后續(xù)都添加進(jìn)去
{
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