hdu 6130 Kolakoski （水题）

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include <string.h>
#include <math.h>
#include <iostream>
#include <stdio.h>
#include <vector>
#include <queue>
using namespace std;
int a[10000009];
int b[10000009];
int sum[10000009];
void init()
{
    a[1]=1;
    a[2]=2;
    b[1]=1;
    b[2]=2;
    sum[1]=1;
    sum[2]=3;
    int k=1,p=3;
    for(int i=3;i<=10000000;i++)
        {if(i>sum[k])
        {
           int t=i-sum[k];
            while(t>b[k+1])
                k++,t-=b[k];
            if(t==1)a[i]=a[i-1]%2+1;
            else a[i]=a[i-1];
        }
        while(p<=i)
        {
            b[p]=a[i];
            sum[p]=sum[p-1]+b[p];
            p++;
        }
        }
}
int main()
{
   int t;
   init();
   scanf("%d",&t);
   while(t--)
   {
       int n;
       scanf("%d",&n);
       printf("%d\n",a[n]);
   }
    return 0;
}