HDU 1005 Number Sequence

http://acm.hdu.edu.cn/showproblem.php?pid=1005

神奇的矩阵

#include <iostream>
using namespace std ;
void mul(int a[2][2],int b[2][2])
{
    int c[2][2] ;
    c[0][0]=(a[0][0]*b[0][0]+a[0][1]*b[1][0])%7 ;
    c[0][1]=(a[0][0]*b[0][1]+a[0][1]*b[1][1])%7 ;
    c[1][0]=(a[1][0]*b[0][0]+a[1][1]*b[1][0])%7 ;
    c[1][1]=(a[1][0]*b[0][1]+a[1][1]*b[1][1])%7 ;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            a[i][j]=c[i][j] ;
}
void qpow(int m[2][2],int n)
{
    int ans[2][2]={1,0,0,1} ;
    int buff[2][2] ;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            buff[i][j]=m[i][j] ;
    while(n)
    {
        if(n&1)
            mul(ans,buff) ;
        mul(buff,buff) ;
        n>>=1 ;
    }
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            m[i][j]=ans[i][j] ;
}
int main()
{
    int a,b,n ;
    while(scanf("%d%d%d",&a,&b,&n),a)
    {
        if(n<=2)
        {
            puts("1") ;
            continue ;
        }
        int m[2][2] ;
        m[0][0]=a%7,m[0][1]=1,
        m[1][0]=b%7,m[1][1]=0 ;
        qpow(m,n-2) ;
        printf("%d\n",(m[0][0]+m[1][0])%7) ;
    }
    return 0 ;
}