2017 Wuhan University Programming Contest 现场赛 I. A simple math problem（矩阵快速幂）

A simple math problem

题目链接：A simple math problem

ps：比赛的时候也想了一下矩阵快速幂，但是不会构造转置矩阵啊。。。。
话说，初始矩阵在转置矩阵里面这一步真的很妙，因为我构造矩阵的时候构造了一个初始矩阵，又构造了个转置矩阵，但是最后答案却是怎样都得不到正确的。。。

代码：

#include<stdio.h>
#include<string.h>

#define N 3
#define mod 11
#define mem(a,b) memset(a,b,sizeof(a))
typedef long long LL;
struct Matrix
{
    LL mat[N][N];
};
Matrix tmp[22];

Matrix operator * (Matrix a,Matrix b)//重载 *
{
    Matrix res;
    for(int i=0; i<N; ++i)
        for(int j=0; j<N; ++j)
        {
            res.mat[i][j]=0;
            for(int k=0; k<N; ++k)
                res.mat[i][j]=(res.mat[i][j]+a.mat[i][k]*b.mat[k][j])%mod;
        }
    return res;
}

Matrix unit_matrix=//单位矩阵
{
    1,0,0,
    0,1,0,
    0,0,1
};

Matrix pow_matrix(Matrix a,LL n)//矩阵快速幂
{
    Matrix res=unit_matrix;
    while(n)
    {
        if(n&1)
            res=res*a;
        a=a*a;
        n>>=1;
    }
    return res;
}

void init()
{
    tmp[1].mat[0][0]=10,tmp[1].mat[1][0]=tmp[1].mat[2][0]=tmp[1].mat[1][1]=tmp[1].mat[2][1]=tmp[1].mat[2][2]=1;
    for(int i=2; i<20; ++i)
    {
        tmp[i]=tmp[i-1];
        tmp[i].mat[0][0]=(tmp[i].mat[0][0]*10)%mod;
    }
}

int main()
{
    LL n;
    Matrix ans;
    scanf("%lld",&n);
    ans=unit_matrix;
    init();
    LL cc=n,len=0;
    while(cc)
    {
        ++len;
        cc/=10;
    }
    LL k=1;
    for(int i=1; i<len; ++i)//不同长度的数有不同的转置矩阵
    {
        ans=ans*pow_matrix(tmp[i],k*9);
        k*=10;
    }
    ans=ans*pow_matrix(tmp[len],n-k+1);
    printf("%lld\n",ans.mat[2][0]);//
    return 0;
}