合并内容流 快手机试题

链接：https://www.nowcoder.com/questionTerminal/0f3bdc60198244d9a8893b843afd2310
来源：牛客网

• 时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 32M，其他语言64M
• 算法知识视频讲解

合并两个内容流，实现隔4个插入1个，如果合并完还有剩下，则加内容流尾部

输入描述:

第行表示第一种类型的内容，字符数量<=100，空格分隔。比如说

1 2 3 4 5 6 7 8 9

a b c

输出描述:

合并两种内容流，输出

1 2 3 4 a 5 6 7 8 b 9 c

示例1

输入

1 2 3 4 5 6 7 8 9
a b c

输出

1 2 3 4 a 5 6 7 8 b 9 c

#include <bits/stdc++.h>
using namespace std;

vector<string> split(string in, string patten)  //字符串分割函数
{
    in += patten;
    vector<string> ve;

    int pos = in.find(patten);
    while (pos != string::npos)
    {
        string tem = in.substr(0, pos);
        ve.push_back(tem);
        in = in.substr(pos+1);
        pos = in.find(patten);
    }

    return ve;
}

int main()
{
    string stra, strb;
    vector<string> vea, veb;

    getline(cin, stra); 
    getline(cin, strb);

    vea = split(stra, " "); //分割字符串
    veb = split(strb, " ");

    int flag = 0;   

    for (int i=0, j=0; i<vea.size(); )
    {
        for (int k=0; k<4; k++) //输出4个a序列字符
        {
            cout<<vea[i++]<<" ";

            if (i >= vea.size())    //a序列先输出完，flag置1，输出剩下的b序列即可
            {
                flag = 1;
                break;
            }
        }
        
        if (j >= veb.size())    //注意不要漏了b序列先输出完的情况，否则会发生段错误！
        {
            for ( ; i<vea.size(); )
            {
                cout<<vea[i++]<<" ";
            }
            break;
        }

        if (flag)   //如果a序列输出完，输出剩下的b序列
        {
            for ( ; j<veb.size(); )
            {
                cout<<veb[j++]<<" ";
            }
        }
        
        else    //注意这里要用else，否则会发生端错误！
            cout<<veb[j++]<<" ";
    }

    return 0;
}