poj 3323 Matrix Power Series （矩阵乘法 非递归形式）

为了搞自动机+矩阵的题目，特来学习矩阵快速幂..........非递归形式的求Sum（A+A^2+...+A^k）不是很懂，继续弄懂................不过代码简洁明了很多，亮神很给力

 

#include <iostream>

#include <algorithm>

#include <cmath>

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <string>

#include <vector>

#include <set>

#include <queue>

#include <stack>

#include <climits>//形如INT_MAX一类的

#define MAX 100005

#define INF 0x7FFFFFFF

#define REP(i,s,t) for(int i=(s);i<=(t);++i)

#define LL long long

#define mem(a,b) memset(a,b,sizeof(a))

#define mp(a,b) make_pair(a,b)

#define L(x) x << 1

#define R(x) x << 1 | 1

# define eps 1e-5

//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂

using namespace std;

int n,k,m;

__int64 a[33][33];

__int64 x[66][66],y[66][66];



void multi(__int64 x[66][66],__int64 y[66][66]) { // A * B

    __int64 p[66][66];

    memset(p,0,sizeof(p));

    int N = n * 2;

    for(int i=0; i<N; i++) {

        for(int j=0; j<N; j++) {

            for(int k=0; k<N; k++) {

                p[i][j] = (p[i][j] + (x[i][k] * y[k][j]) % m) % m;

            }

        }

    }

    for(int i=0; i<N; i++) {

        for(int j=0; j<N; j++) {

            x[i][j] = p[i][j];

        }

    }

}



void quickmul(int p) { //将矩阵扩大成2n * 2n

    for(int i=0; i<n; i++) {

        for(int j=0; j<n; j++) {

            y[i+n][j+n] = a[i][j];

            x[i][j+n] = a[i][j];

        }

    }

    for(int i=0; i<n; i++) y[i][i] = 1;

    for(int i=0; i<n; i++) y[i+n][i] = 1;

    while(p) {  //A ^ p

        if(p & 1) {

            multi(x,y);

        }

        multi(y,y);

        p = p >> 1;

    }

}



int main() {

    scanf("%d%d%d",&n,&k,&m);

    for(int i=0; i<n; i++) {

        for(int j=0; j<n; j++) {

            scanf("%d",&a[i][j]);

            a[i][j] %= m;

        }

    }

    memset(x,0,sizeof(x));

    memset(y,0,sizeof(y));

    quickmul(k);

    for(int i=0; i<n; i++) {

        printf("%d",x[i][0]);

        for(int j=1; j<n; j++) printf(" %d",x[i][j]);

        puts("");

    }

    return 0;

}