hdu 1005 Number Sequence（矩阵连乘+二分快速求幂）

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

代码：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int mod=7;
struct matrix
{
  int f[2][2];
};

matrix mul(matrix a,matrix b)
{
    int i,j,k;
    matrix c;
    memset(c.f,0,sizeof(c.f));
    for(i=0;i<2;i++)
    {
        for(j=0;j<2;j++)
        {
            for(k=0;k<2;k++)
            {
                c.f[i][j]+=a.f[i][k]*b.f[k][j];
                c.f[i][j]%=mod;
            }
        }
    }
    return c;
}

matrix pow_mod(matrix a,int b)
{
    matrix s;
    int t;
    memset(s.f,0,sizeof(s.f));
    for(int i=0;i<2;i++)
    {
        s.f[i][i]=1;
    }
    while(b)
    {
        /*if(b&1)
        {
            s=mul(s,a);
        }
        a=mul(a,a);
        b=b>>1;*/
        t=b%2;
        b/=2;
        if(t!=0)
            s=mul(s,a);
         a=mul(a,a);
    }
    return s;
}
int main()
{
    int a,b,n;
    while(scanf("%d%d%d",&a,&b,&n))
    {
        if(a==0&&b==0&&n==0)
            break;
         matrix e;
    e.f[0][0]=a;
    e.f[0][1]=1;
    e.f[1][0]=b;
    e.f[1][1]=0;
    e=pow_mod(e,n-1);
    printf("%d
",(e.f[0][1]+e.f[1][1])%mod);
    }

    return 0;
}