HDU_6130 Kolakoski 【规律】

题目链接

题目描述

就是一个Kolakoski序列，让你求其通项公式。给n求a[n]即可

解题思路

它的定义很简单，若把数列中相同的数定为一组，令a(1)=1，a(2)=2，则a(n)等于第n组数的长度。
可以根据这个定义来推算第三项以后的数：例如由于a(2)=2，因此第2组数的长度是2，因此a(3)=2,；
由于a(3)=2，所以第三组数的长度是2，因此a(4)=a(5)=1；由于a(4)=1，a(5)=1，所以第四组数和第五组数的长度都为1，因此a(6)=2，a(7)=1，以此类推。

代码部分

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
const int maxn = 1e7 + 10;
int a[maxn];
int main()
{
    ios::sync_with_stdio(false);
    a[0] = 1;
    a[1] = 2;
    a[2] = 2;
    int cnt = 3;
    int flag = 1, ans = 2;
    for(int i = 2; cnt <= maxn + 10; ++ i)
    {
        ans += flag * (-1);
        flag *= (-1);
        for(int j = 1; j <= a[i]; ++ j)
        {
            a[cnt] = ans;
            cnt ++;
        }

    }
    int t;
    cin >> t;
    while(t --)
    {
        int n;
        cin >> n;
        cout << a[n - 1] << endl;
    }
    return 0;
}