Kolakoski序列产生器

/*
本程序说明：

Kolakoski序列是一个仅由1和2组成的无限数列，是一种通过“自描述”来定义的数列。
他的前几项为1,2,2,1,1,2,1,2,2,1,2,2,1,1,2,1,1,2,2,1,2,
如果相同数字是一组，则每一组的个数为1,2,2,1,1,2,1,2,2,1,2,2,1,1
a(n)等于第n组数的长度，因此是自描述的。目前没有递推公式，只能用自描述方式给出
本题为Kolakoski序列的变形，可输入任意一组数，构建Kolakoski序列
输入为 n m. n是长度，m是给的数个数，剩下一行是数组.
 20 2
 1 2
 输出：1 2 2 1 1 2 1 2 2 1 2 2 1 1 2 1 1 2 2 1
 30 4
 2 1 3 1
 输出：2 2 1 1 3 1 2 2 2 1 3 3 1 1 2 2 1 3 3 3 1 1 1 2 1 3 3 1 1 2

*/
#include <iostream>
#include <vector>
using namespace std;

int main()
{
    int n,m;
    while(cin>>n>>m){
        vector<int> v(m);
        for(int i=0;i<m;++i){
            cin>>v[i];
        }
        vector<int> res;

        int i=0;
        int j=0;
        int tmp=v[0];//启动项
        int count=0;
        while(count<n){
            while(tmp--){
                res.push_back(v[i]);
                ++count;
                if(count==n)
                    break;
            }
            ++i;
//            if(i==(int)v.size())//v到达尾部，重新置为头部开始循环
//                i=0;
            //或
            i%=(int)v.size();

            ++j;
            tmp=(res.size()<2)?v[1]:res[j];//防止res位数不够
        }
        for(int val:res)
            cout<<val<<" ";
    }
    return 0;
}

 原题截图：