合并两个有序数组（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());
}