UVA 11149 Power of Matrix 倍增法

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn 44
#define mod 10
int n;
struct matrix{
    int f[maxn][maxn];
};
matrix sum(matrix a,matrix b)
{
    int i,j;
    matrix s;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            s.f[i][j]=(a.f[i][j]+b.f[i][j])%mod;
    return s;
}
matrix mul(matrix a,matrix b)
{
    int i,j,k;
    matrix s;
    memset(s.f,0,sizeof(s.f));
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            for(k=0;k<n;k++)
            s.f[i][j]=(s.f[i][j]+a.f[i][k]*b.f[k][j])%mod;
    return s;
}
matrix pows(matrix e,int b)
{
    matrix s,a;
    int i,j,k;
    a=e;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        if(i==j)s.f[i][j]=1;
        else s.f[i][j]=0;
    while(b)
    {
        if(b&1)
            s=mul(s,a);
        a=mul(a,a);
        b=b>>1;
    }
    return s;
}
matrix work(matrix e,int k)
{
    matrix s,a,b;
    if(k==1)
        return e;
    a=work(e,k/2);
    s=sum(a,mul(a,pows(e,k/2)));
    if(k&1)
        s=sum(s,pows(e,k));
    return s;
}
int main()
{
    int k;
    while(cin>>n>>k)
    {
        if(n==0)//坑啊，习惯的写成if(n==0&&k==0)break;。。。
            break;
        int i,j,a;
        matrix e;
        for(i=0;i<n;i++)
        {
            for(j=0;j<n;j++)
            {
                cin>>a;
                e.f[i][j]=a%10;
            }
        }
        e=work(e,k);
        for(i=0;i<n;i++)
        {
            cout<<e.f[i][0];
            for(j=1;j<n;j++)
                cout<<" "<<e.f[i][j];
            cout<<endl;
        }
        cout<<endl;
    }
    return 0;
}
/*
    倍增法
    a+a^2+a^3+a^4+a^5=(a+a^2)+a^2*(a+a^2)+a^5(5为奇数),a+a^2=a+a*(a)(2为偶数)
    
    或则k=100=1010B，s=0
    从左到右，当为1，s=s*2+1;否则s=s*2
    a+a^2+a^3+...+a^10:(个数变化)
    1:a
    0:a+a^2=a+a*(a);
    1:a+a^2+a^3+a^4+a^5=(a+a^2)+a^2*(a+a^2)+a^5
    0:a+a^2+...^a^10=(a+a^2+...+a^5)+a^5*(a+a^2+...+a^5)
*/