UVA-10689 Yet another Number Sequence （矩阵二分幂模板）

题目大意：已知递推公式和边缘值，求某项的最后m(0<m<5)位数字。

题目分析：矩阵二分幂的模板题。

代码如下：

# include<iostream>
# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
struct matrix
{
    int r,c,m[3][3];
    matrix(int _r,int _c):r(_r),c(_c){}
};
int a,b,n,m;
int mod[4]={10,100,1000,10000};
matrix multiply(matrix a,matrix b)
{
    matrix c(a.r,b.c);
    for(int i=1;i<=c.r;++i){
        for(int j=1;j<=c.c;++j){
            c.m[i][j]=0;
            for(int k=1;k<=a.c;++k){
                c.m[i][j]+=(a.m[i][k]*b.m[k][j]);
                c.m[i][j]%=mod[m-1];
            }
        }
    }
    return c;
}
matrix matrix_pow(matrix a,int k)
{
    if(k==0){
        for(int i=1;i<=a.r;++i)
            for(int j=1;j<=a.c;++j)
                a.m[i][j]=(i==j)?1:0;
        return a;
    }
    if(k==1)
        return a;
    matrix res=matrix_pow(a,k/2);
    res=multiply(res,res);
    if(k&1)
        res=multiply(res,a);
    return res;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d%d",&a,&b,&n,&m);
        if(n==0){
            printf("%d
",a%mod[m-1]);
            continue;
        }
        matrix mat(2,2);
        mat.m[1][1]=mat.m[1][2]=mat.m[2][1]=1,mat.m[2][2]=0;
        mat=matrix_pow(mat,n-1);
        matrix ans(2,1);
        ans.m[1][1]=b,ans.m[2][1]=a;
        ans=multiply(mat,ans);
        printf("%d
",ans.m[1][1]);
    }
    return 0;
}