Number Sequence hdu 构造矩阵乘法

/*题目所给的a，b和f的初始值很重要。
一次来确定矩阵的元素和幂。
[a b1 0]×[f(n−1)f(n−2)]=[f(n)f(n−1)]
[a b1 0]n−2×[11]=[f(n)f(n−1)]
因为f[1]=1,f[2]=1,
所以递推到n-2次幂，间接矩阵为1,1.
如果初始条件改变的话，次幂也间接矩阵也会随之改变。
要灵活应用。*/
#include <stdio.h>
#include <cstring>
const int mod=7;
int A[3][3],ans[3][3];
void mul(int a[][3],int b[][3])
{
    int tmp[3][3];
    for(int i=1; i<=2; i++)
        for(int j=1; j<=2; j++)
        {
            tmp[i][j]=0;
            for(int k=1; k<=2; k++)            
                tmp[i][j]+=a[i][k]*b[k][j];            
            if(tmp[i][j]>=mod) tmp[i][j]%=mod;
        }
    memcpy(a,tmp,sizeof(tmp));
}
void pow(int a[][3],int b[][3],int n)
{   
    a[1][1]=a[2][2]=1;
    a[1][2]=a[2][1]=0;
    while(n)
    {
        if(n&1)//n为奇数的话会先执行一次，最后n必为1，又执行一次，则将结果都传递到数组a上。
            mul(a,b);
        mul(b,b);
        n>>=1;
    }
}
int main()
{
    int n,a,b;
    while(scanf("%d%d%d",&a,&b,&n)==3)
    {
        if(a==0&&b==0&&n==0) break;
        if(n<=2) printf("1\n");
        else
        {
            memset(ans,0,sizeof(ans));
            A[1][1]=a;
            A[1][2]=b;
            A[2][1]=1;
            A[2][2]=0;
            pow(ans,A,n-2);
            printf("%d\n",(ans[1][2]+ans[1][1])%7);
        }
    }
    return 0;
}