NYOJ 427 & HDU 1005 Number Sequence

题目链接：Click here~~

f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.

这也是月赛的一道题，题意很好理解，就是按照上面的公式求出f(n)。由于n很大，单靠循环做肯定会超时。我首先想到了用矩阵二分幂做。

结果我推导那个矩阵用了很久，好不容易推导出来，结果还是wa了，由于矩阵知识掌握的并不牢固，所以我也没信心继续调试了，然后我就收手去看别的题了。

比赛结束后，看了看解题报告，明白了此题可以用找循环节的方法来做，因为f(n)的范围是0~6,也就是只有7种情况，所以连续两个数f(n-1)f(n)的情况最多有7*7=49种。而且f(n)只取决于f(n-1)和f(n-2)的大小，所以一定会有循环。所以只要找到循环开始的位置和循环节的长度，此题就迎刃而解了。

#include <stdio.h>
#include <string.h>
int in[7][7];   //表示连续的数字为a、b时,a出现的位置
int main()
{
    int a,b,n,f[53]={0,1,1},circle,from;
    while(scanf("%d%d%d",&a,&b,&n),a)
    {
        a%=7 , b%=7;
        memset(in,0,sizeof(in));
        in[1][1]=1;
        for(int i=3;i<=51;i++)
        {
            f[i]=(a*f[i-1] + b*f[i-2])%7;
            if(in[ f[i-1] ][ f[i] ])
            {
                circle = i-1 - in[ f[i-1] ][ f[i] ];    //循环节长度
                from   = in[ f[i-1] ][ f[i] ] ;         //循环开始的位置
                break;
            }
            in[ f[i-1] ][ f[i] ] = i-1;
        }
        printf("%d\n",f[ from + (n-from)%circle ]);
    }
    return 0;
}