Power of Matrix UVA - 11149 (矩阵快速幂)

题目链接 ： 点击
这个题就是求S=A+A^2+A^3+A^4……
题的最后一句： Since the values may be very large, you only need to print their last digit. 保留最后一位数

//倍增法
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<fstream>
#include<ctype.h>
#include<math.h>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<string>
#include<iostream>
#include<algorithm>
#include<utility>
#include<iomanip>
#include<time.h>
#include<iostream>
#define lowbit(x) (x&-x)
#define abs(x) ((x)>0?(x):-(x))
using namespace std;
typedef long long ll;
const double Pi = acos(-1.0);
const int N = 1e6+10, M = 1e3+20, mod = 1e9+7, inf = 2e9+10;
const double e=2.718281828459 ;
const double esp=1e-9;
int n,k;
struct Matrix
{
    int mat[44][44];
    Matrix()
    {
        memset(mat,0,sizeof(mat));
        for(int i=0; i<44; i++) mat[i][i]=1;
    }
} E;
Matrix Multi(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])%10;
        }
    }
    return res;
}
Matrix Add(Matrix a, Matrix b)
{
    Matrix res;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            res.mat[i][j]=(a.mat[i][j]+b.mat[i][j])%10;
        }
    }
    return res;
}
Matrix Pow(Matrix a,int n)
{
    Matrix res;
    while(n)
    {
        if(n&1) res=Multi(res,a);
        a=Multi(a,a);
        n>>=1;
    }
    return res;
}
Matrix Get_Ans(Matrix a,int k)
{
    if(k==1) return a;
    if(k&1)  return Add(Pow(a,k),Get_Ans(a,k-1));
    if(k%2==0)
    {
        Matrix A=Get_Ans(a,k/2);
        Matrix B=Pow(a,k/2);
        Matrix C=Multi(A,B);
        return Add(C,A);
    }
}
int main()
{
    Matrix A;
    while(~scanf("%d%d",&n,&k))
    {
        if(n==0) break;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                scanf("%d",&A.mat[i][j]);
                A.mat[i][j]%=10;
            }
        }
        Matrix ans=Get_Ans(A,k);
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                if(j) printf(" ");
                printf("%d",ans.mat[i][j]);
            }
            printf("\n");
        }
        printf("\n");
    }
    return 0;
}