POJ 3233（二分求和+矩阵快速幂）

二分求和：

/*
矩阵快速幂
*/

#include<iostream>
#include<cstdio>

using namespace std;

const int maxn=35;//刚开始开成100 一直RE 无语

typedef struct
{
    int m[maxn][maxn];
}Matrax;

Matrax a,per;

int n,M;

void init()
{
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
            scanf("%d",&a.m[i][j]);
            a.m[i][j]%=M;
            per.m[i][j]=(i==j);
        }
}

Matrax multi(Matrax a,Matrax b)
{
    Matrax c;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
            c.m[i][j]=0;
            for(int k=0;k<n;k++)
                c.m[i][j]+=a.m[i][k]*b.m[k][j];
            c.m[i][j]%=M;
        }
    return c;
}

Matrax power(int k)
{
    Matrax c,p,ans=per;
    p=a;
    while(k){
        if(k&1)
            ans=multi(ans,p);
        k>>=1;
        p=multi(p,p);
    }
    return ans;
}

Matrax add(Matrax a,Matrax b)
{
    Matrax c;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
            c.m[i][j]=(a.m[i][j]+b.m[i][j])%M;
        }
    return c;
}

//二分求和
Matrax Matraxsum(int k)
{
    if(k==1)
        return a;
    Matrax temp,b;
    temp=Matraxsum(k/2);
    if(k&1){
        b=power(k/2+1);
        temp=add(temp,multi(temp,b));
        temp=add(temp,b);
    }
    else{
        b=power(k/2);
        temp=add(temp,multi(temp,b));
    }
    return temp;
}

int main()
{
    int k;
    while(scanf("%d%d%d",&n,&k,&M)!=EOF){
        init();
        Matrax ans=Matraxsum(k);
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                printf("%d ",ans.m[i][j]);
            }
            printf("\n");
        }

    }
    return 0;
}