UVA-11149 Power of Matrix

倍增法求矩阵幂的和：https://blog.csdn.net/luomingjun12315/article/details/50824552

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const ll MOD=10;
int N;
struct node
{
	ll a[45][45];
};
node shu,ans,mp;
//shu是输入的矩阵，ans是所求答案
node matrix(node x,node y)
{
    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++){
            mp.a[i][j]=0;
            for(int p=1;p<=N;p++)
                mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]+MOD)%MOD;
            //矩阵乘法 
        }
    return mp;
}
void work(ll k)
{//矩阵快速幂 
	for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++)
				ans.a[i][j]=0;
	for(int i=1;i<=N;i++) ans.a[i][i]=1;
	node t=shu;
    while(k){
        if(k&1) 
            ans=matrix(ans,t);
        k>>=1;
        t=matrix(t,t);
    }
}
node solve(node a,int n) //倍增法求和
{
	if(n==1) return a;
	node t=solve(a,n/2);
	shu=a;
	work(n/2);
	node p=matrix(t,ans);
	for(int i=1;i<=N;i++)
		for(int j=1;j<=N;j++)
			t.a[i][j]=(t.a[i][j]+p.a[i][j])%MOD;
	if(n&1)
	{
		shu=a;
		work(n);
		for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++)
				t.a[i][j]=(t.a[i][j]+ans.a[i][j])%MOD;
	}
	return t;
}
node a;
int main()
{
	ll n,m;
	while(~scanf("%lld%lld",&n,&m))
	{
		if(n==0) break;
		N=n;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				scanf("%lld",&a.a[i][j]),a.a[i][j]%=MOD; //这里必须先取余，否则WA，可能溢出了
		node t=solve(a,m);
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				printf("%lld",t.a[i][j]);
				if(j==n) printf("\n");
				else printf(" ");
			}
		}
		printf("\n");
	}
	return 0;
}