ACM模版

描述

题解

一看公式就知道这道题在51Nod上做过一次,于是按照老思路准备水过,可是却意外发现了自己曾经的写法实在是想当然了,如果不是这道题51Nod数据比较水,我一定过不去~~~

以前在做这道题时,感觉循环一定是从第一项开始的,也就是循环节的前两项一定是1、1,然而事实并非如此,当我依然用这种思路写时,HDU给我的结果是WA,于是我改掉了这个可能存在的问题,结果顺利AC了……

看来有一句话说的真不假,你样例过了,并不能代表你对了,甚至你AC了,同样不能确保你是对的!!!

代码

#include <iostream>

using namespace std;

const int MAXN = 300;
const int MOD = 7;

int f[MAXN] = {
  1, 1, 1};

int main(int argc, const char * argv[])
{
    int A, B, n;
    while (cin >> A >> B >> n)
    {
        if (A == 0 && B == 0 && n == 0)
        {
            break;
        }

        int beg = 1;
        int end = 1;
        int cycle = 1;
        bool flag = false;
        for (int i = 3; i <= n && !flag; i++)
        {
            f[i] = (A * f[i - 1] + B * f[i - 2]) % MOD;
            for (int j = 2; j < i; j++)
            {
                if (f[i] == f[j] && f[i - 1] == f[j - 1])
                {
                    cycle = i - j;
                    beg = j - 1;
                    end = i - 1;
                    flag = true;
                    break;
                }
            }
        }

        if (flag)
        {
            cout << f[beg + (n - end) % cycle] << '\n';
        }
        else
        {
            cout << f[n] << '\n';
        }
    }

    return 0;
}

参考

51Nod 1126 求递推序列的第N项