HDU 1005 Number Sequence

题目链接:http://acm.hdu.edu.cn/showproblem.php?

pid=1005

找周期。当f(n)函数值再次出现1。1的时候就是一个循环。

#include<cstdio>
using namespace std;
#define maxn 1000+5
int main()
{
    int a,b,n,i,f[maxn];
    f[1]=1;
    f[2]=1;
    while(~scanf("%d%d%d",&a,&b,&n))
    {
        if(a==0&&b==0&&n==0)
            break;
        else
        {
            for(i=3; i<=1000; i++)
            {
                f[i]=(a*f[i-1]+b*f[i-2])%7;
                if(f[i]==1&&f[i-1]==1)
                    break;//i-2为周期
            }
            f[0]=f[i-2];//这里注意当n为i-2的倍数时。n对i-2取余为0。而之前并没有对f[0]赋值。所以一定要把f[0]赋为f[i-2]。
            n=n%(i-2);
            printf("%d
",f[n]);
        }
    }
    return 0;
}