HRBUST 1126 Final Destination II （矩阵快速幂）

题目：http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1126

代码：

#include<stdio.h>
#include<string.h>
#define MOD 1000000007

using namespace std;

struct node
{
   long long int m[3][3];
}ans,base;

node mliti(node a,node b)
{
    node temp;

    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
        {
            temp.m[i][j]=0;
            for(int k=0;k<3;k++)
            {
                temp.m[i][j]=(temp.m[i][j]+a.m[i][k]*b.m[k][j])%MOD;
            }
        }
    }
    return temp;
}

long long int fast(long long int n)
{
    ans.m[1][1]=ans.m[0][0]=ans.m[2][2]=1;
    ans.m[0][1]=ans.m[0][2]=ans.m[1][0]=ans.m[1][2]=ans.m[2][0]=ans.m[2][1]=0;

    base.m[0][0]=base.m[0][1]=base.m[0][2]=base.m[1][0]=base.m[2][1]=1;
    base.m[1][1]=base.m[1][2]=base.m[2][0]=base.m[2][2]=0;

    while(n)
    {
        if(n%2==1)
            ans=mliti(ans,base);
        base=mliti(base,base);
        n=n/2;
    }
    return ans.m[0][0];
}

int main()
{
    int ui=0;
    int t;
    scanf("%d",&t);
    while(t--)
    {
        long long int n;
        scanf("%lld",&n);
        printf("Case %d:\n%lld\n",++ui,fast(n));
    }
}