合並兩個有序數組（C++）

合並兩個有序數組

思路

• 存在一個數組為空，則直接返回另一個（非）空數組。
• 都不為空

兩個數組都非空：記兩個數組分別為v1,v2，合並后的數組為vc，設置兩個指針i、j分別指向數組v1,v2，令i=j=0，

若v1[i] <= v2[j]，則可以一直添加v1的元素到合並后的數組vc直到條件不滿足,同時修改i；

添加v2的元素到vc；

如果一個數組被遍歷完，則應該終止；

添加未遍歷完的數組的其余元素到vc。

代碼

vector<int> merge(vector<int> &v1, vector<int> &v2)
{
    if(v1.empty()) return v2;
    if(v2.empty()) return v1;
    
    vector<int> r;
    unsigned int i = 0,j = 0;
    while(i < v1.size() && j < v2.size())
    {
        while(v1[i]<=v2[j] && i < v1.size())
        {
            r.push_back(v1[i]);
            i++;
        }
        while(v1[i]>=v2[j] && j < v2.size())
        {
            r.push_back(v2[j]);
            j++;
        }
    }
    while(i<v1.size()) 
    {
        r.push_back(v1[i]);
        i++;
    }
    while(j<v2.size()) 
    {
        r.push_back(v2[j]);
        j++;
    }
    
    return r;
}

 

測試代碼

#include <iostream>
#include <vector>
#include <cassert>
using namespace std;
void printVec(const vector<int> &v)
{
    for(auto &x: v)
        cout << x << ' ';
    cout << endl;
}

// 每次遍歷使當前元素最小，第二次為次最小,...
vector<int> sortAscend(vector<int> &v)
{
    vector<int> r = v;
    for(size_t i = 0; i < r.size(); i++)
    {
        for(size_t j = i+1; j < r.size(); j++)
        {
            if(r[i] > r[j])
            {
                int t = r[j];
                r[j] = r[i];
                r[i] = t;
            }
        }
        // printVec(r);
    }
    return r;
}
 
int main()
{
    std::vector<int> v1 = {7, 5, 16, 5, 8, 1,2}, v1r;
    std::vector<int> v2 = {3,3,6,3,0,1,0}, v2r;
    v1r = sortAscend(v1);
    cout << "-----------------------------" << endl;
    v2r = sortAscend(v2);
    vector<int> r = merge(v1r, v2r);
    printVec(v1r);
    printVec(v2r);
    printVec(r);
    assert(r.size() == v1r.size()+v2r.size());
}