poj 3323 Matrix Power Series (矩阵乘法 非递归形式)

来源:互联网 发布:新买的被子要洗吗 知乎 编辑:程序博客网 时间:2024/05/18 20:05

为了搞自动机+矩阵的题目,特来学习矩阵快速幂..........非递归形式的求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;}