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是给的数个数，剩下一行是数组.
// 10 2
// 1 2
// 输出：1 2 2 1 1 2 1 2 2 1 
#include <iostream>
#include <vector>
using namespace std;

int main()
{
	int n, m;
	cin >> n >> m;
	vector<int> p(m);
	for (int i = 0; i < m; i++)
		cin >> p[i];
	int k = 0, j = 0;
	vector<int> t;
	while (j < n){
		int a = p[k%m];
		int b = k == j ? a : t[k];
		for (int i = 0; i < b; i++){
			t.push_back(a); 
			j++;
		}	
		k++;
	}
	for (int i = 0; i < n; i++)
		cout << t[i] << endl;
	return 0;
}