题目传送门:http://acm.hdu.edu.cn/showproblem.php?pid=6130
Kolakoski序列是一个仅由1和2组成的无限数列,是一种通过“自描述”来定义的数列[1] 。他的前几项为
1,2,2,1,1,2,1,2,2,1,2,2,1,1,2,1,1,2,2,1,2,1,1,2,1,2,2,1,1,…
它的定义很简单,若把数列中相同的数定为一组,令a(1)=1,a(2)=2,则a(n)等于第n组数的长度。
打表直接求
k表示的前面那一组的值,相邻两组不相等
a[j]就是所求组的个数,1个还是2个
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#define N 10000005
using namespace std;
int a[N];
void init()
{
a[1] = 1;
a[2] = 2;
a[3] = 2;
int k = 2;
for(int i = 4, j = 3; i < N; i++)
{
if(a[j] == 1)
{
if(k == 1)
a[i] = 2, k = 2;
else
a[i] = 1, k = 1;
}
else
{
if(k == 1)
a[i] = 2, a[++i] = 2, k = 2;
else
a[i] = 1, a[++i] = 1, k = 1;
}
j ++;
}
}
int main()
{
init();
int T;
scanf("%d", &T);
while(T--)
{
int b;
scanf("%d", &b);
printf("%d\n", a[b]);
}
return 0;
}