POJ 3233 Matrix Power Series（矩阵快速幂+二分）

Description
给出一个n*n矩阵，求矩阵S=A+A^2+A^3+…+A^k(mod m)
Input
第一行包括三个整数n,m,k,之后为一n*n矩阵
Output
输出S
Sample Input
2 2 4
0 1
1 1
Sample Output
1 2
2 3
Solution1
令，所以只要构造一个2n*2n的矩阵B，然后用矩阵快速幂算出B^(n+1)之后取B的右上角矩阵，然后减去一个单位矩阵即为答案
Code1

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 123 
typedef long long ll;
struct Mat
{
    int mat[maxn][maxn];//矩阵 
    int row,col;//矩阵行列数 
};
Mat mod_mul(Mat a,Mat b,int p)//矩阵乘法 
{
    Mat ans;
    ans.row=a.row;
    ans.col=b.col;
    memset(ans.mat,0,sizeof(ans.mat));
    for(int i=0;i<ans.row;i++)
        for(int j=0;j<ans.col;j++)
            for(int k=0;k<a.col;k++)
            {
                ans.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
                ans.mat[i][j]%=p;
            }
    return ans;
}
Mat mod_pow(Mat a,int k,int p)//矩阵快速幂 
{
    Mat ans;
    ans.row=a.row;
    ans.col=a.col;
    for(int i=0;i<a.row;i++)
        for(int j=0;j<a.col;j++)
            ans.mat[i][j]=(i==j);
    while(k)
    {
        if(k&1)ans=mod_mul(ans,a,p);
        a=mod_mul(a,a,p);
        k>>=1;
    }
    return ans;
}
int main()
{
    int n,m,k;
    while(scanf("%d%d%d",&n,&k,&m)!=EOF)
    {
        Mat A;
        A.row=A.col=2*n;
        memset(A.mat,0,sizeof(A.mat));
        //构造B矩阵 
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&A.mat[i][j]);
        for(int i=0;i<n;i++)
            A.mat[i][i+n]=A.mat[i+n][i+n]=1;
        Mat B=mod_pow(A,k+1,m);
        for(int i=0;i<n;i++)//取右上角部分减去一单位矩阵 
            for(int j=n;j<2*n;j++)
            {
                if(i+n==j)printf("%d",((B.mat[i][j]-1)%m+m)%m);//注意取余 
                else printf("%d",B.mat[i][j]);
                printf("%c",j==2*n-1?'\n':' ');
            }
    }
    return 0;
} 

Solution2

递归求解，每次问题规模折半，时间复杂度
Code2

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 33
struct Mat
{
    int mat[maxn][maxn];//矩阵 
    int row,col;//矩阵行列数 
};
Mat Sum(Mat a,Mat b,int p)//矩阵加法 
{
    Mat ans;
    ans.row=a.row,ans.col=a.col;
    for(int i=0;i<a.row;i++)
        for(int j=0;j<a.col;j++)
            ans.mat[i][j]=(a.mat[i][j]+b.mat[i][j])%p;
    return ans; 
}
Mat Mul(Mat a,Mat b,int p)//矩阵乘法 
{
    Mat ans;
    ans.row=a.row;
    ans.col=b.col;
    memset(ans.mat,0,sizeof(ans.mat));
    for(int i=0;i<ans.row;i++)      
        for(int k=0;k<a.col;k++)
            if(a.mat[i][k])
                for(int j=0;j<ans.col;j++)
                {
                    ans.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
                    ans.mat[i][j]%=p;
                }
    return ans;
}
Mat Pow(Mat a,int k,int p)//矩阵快速幂 
{
    Mat ans;
    ans.row=a.row;
    ans.col=a.col;
    for(int i=0;i<a.row;i++)
        for(int j=0;j<a.col;j++)
            ans.mat[i][j]=(i==j);
    while(k)
    {
        if(k&1)ans=Mul(ans,a,p);
        a=Mul(a,a,p);
        k>>=1;
    }
    return ans;
}
Mat A,I;
Mat Solve(int k,int p)
{
    if(k==1)return A;
    if(k&1)return Sum(Pow(A,k,p),Mul(Sum(I,Pow(A,k/2,p),p),Solve(k/2,p),p),p);
    return Mul(Sum(I,Pow(A,k/2,p),p),Solve(k/2,p),p);
}
int main()
{
    int n,k,m;
    while(~scanf("%d%d%d",&n,&k,&m))
    {
        A.col=A.row=n;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&A.mat[i][j]);
        I.col=I.row=n;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                I.mat[i][j]=(i==j);
        A=Solve(k,m);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                printf("%d%c",A.mat[i][j],j==n-1?'\n':' ');
    }
    return 0;
}