hdu 1005 矩阵快速幂

题目传送门

构造转移矩阵，进行矩阵快速幂即可，注意特判n<3时的情况

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long LL;
int mod=7,a,b;
struct matrix{
int m[2][2];
};
matrix mul(matrix A,matrix B)
{
    matrix ret;
    ret.m[0][0]=(A.m[0][0]*B.m[0][0]+A.m[0][1]*B.m[1][0])%mod;
    ret.m[0][1]=(A.m[0][0]*B.m[0][1]+A.m[0][1]*B.m[1][1])%mod;
    ret.m[1][0]=(A.m[1][0]*B.m[0][0]+A.m[1][1]*B.m[1][0])%mod;
    ret.m[1][1]=(A.m[1][0]*B.m[0][1]+A.m[1][1]*B.m[1][1])%mod;
    return ret;
}
matrix pow(matrix A,LL n)
{
    matrix ret;
    ret.m[0][0]=1,ret.m[0][1]=0,ret.m[1][0]=0,ret.m[1][1]=1;
    while(n)
    {
        if(n&1) ret=mul(ret,A);
        A=mul(A,A);
        n>>=1;
    }
    return ret;
}
int main()
{
    int n;
    while(scanf("%d%d%d",&a,&b,&n)==3)
    {
        if(!a&&!b&&!n) break;
        if(n<=2) {
            printf("1\n");
            continue;
        }
        matrix A;
        A.m[0][0]=a;
        A.m[0][1]=b;
        A.m[1][0]=1;
        A.m[1][1]=0;
        matrix ans1=pow(A,n-2);
        printf("%d\n",(ans1.m[0][0]+ans1.m[0][1])%mod);
    }
    return 0;
}