HDU 6130 Kolakoski

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6130
求序列的第n个元素。
解释：
https://baike.baidu.com/item/Kolakoski%E5%BA%8F%E5%88%97/16683856?fr=aladdin
其实就是：
1 2 3 4 5 6 7 8 9
1 2 2 1 1 2 1 2 2
决定长度 1，2轮换
a[3]是2，所以a[4],a[5]都为1；
a[4]是1，所以a[6]是2；
a[5]是1，所以a[7]是1；
a[6]是2，所以a[8],a[9]都是2；

#include<bits/stdc++.h>
using namespace std;
int a[10000005];
int main()
{
    memset(a,0,sizeof(a));
    a[1]=1;
    a[2]=a[3]=2;
    int s=1,j=4,k=3,i,t,n;
    while(j<=10000000)
    {
        for(i=1; i<=a[k]; i++)
        {
            a[j++]=s;
        }
        k++;
        s=3-s;
    }
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        printf("%d\n",a[n]);
    }
}