HDU1575 矩阵快速幂入门

Problem Description
A为一个方阵，则Tr A表示A的迹（就是主对角线上各项的和），现要求Tr(A^k)%9973。

Input
数据的第一行是一个T，表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行，每行有n个数据，每个数据的范围是[0,9]，表示方阵A的内容。

Output
对应每组数据，输出Tr(A^k)%9973。

Sample Input
2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9

Sample Output
2
2686

其实就和快速幂一样…道理不难
但是不太好写…
特别是开始对jieguo这个变量赋值要分各种情况…
巨烦…
过几天去掉一些东西把这个玩意装进函数里
会改成模板的….

#include<iostream>#include<cstdio>#include<queue>#include<algorithm>#include<memory.h>#include<cstdlib>using namespace std;struct p{    int tu[11][11];    void ini()    {        for(int a=0;a<=10;a++)        {            for(int b=0;b<=10;b++)            {                tu[a][b]=0;            }        }    }    void iny()    {        for(int a=0;a<=10;a++)        {            for(int b=0;b<=10;b++)            {                tu[a][b]=1;            }        }    }};int n,k;p xiangcheng(p a,p b){    p c;    c.ini();    for(int q=1;q<=n;q++)    {        for(int w=1;w<=n;w++)        {            for(int e=1;e<=n;e++)            {                c.tu[q][w]+=a.tu[q][e]*b.tu[e][w];                c.tu[q][w]%=9973;            }        }    }    return c;}p q,jieguo,w;int main(){    int T;    cin>>T;    memset(&q,0,sizeof(&q));    while(T--)    {        cin>>n>>k;        q.ini();        w.ini();        jieguo.ini();        int qq=k;        for(int a=1;a<=n;a++)for(int b=1;b<=n;b++)cin>>q.tu[a][b];        w=q;        if(qq&1)qq>>=1,jieguo=w,w=xiangcheng(w,w);        else        {            while((qq&1)==0)            {                w=xiangcheng(w,w);                qq>>=1;            }            qq>>=1;            jieguo=w;            w=xiangcheng(w,w);        }        while(qq)        {            if(qq&1)jieguo=xiangcheng(jieguo,w);            w=xiangcheng(w,w);            qq>>=1;        }        int ans=0;        for(int a=1;a<=n;a++)        {            ans+=jieguo.tu[a][a];            ans%=9973;        }        cout<<ans<<endl;    }    return 0;}