poj 3233 Matrix Power Series

//非常经典的题目,涉及两次二分

//方法:A + A^2 + A^3 + A^4 + A^5 + A^6 =(A + A^2 + A^3) + A^3*(A + A^2 + A^3)

//开始的时候写的递归,果断TLE,改成迭代,还TLE ,,仔细看过后发现自己在许多不需要的地方取模了(证明取模是多么的慢)

//最后改成直接求幂而不用使用pow()函数的,提速了10倍

//还值得提一句的就是,,二进制数序列用于保存路径的确很好用 : )

#include <iostream> #include <queue> #include <string> #include <map> #include <vector> #include <cstring> #include <cstdio> #include <cmath> #include <algorithm> using namespace std; const int MAX=35,INF=1<<30; int n,m,k; int vec[MAX],l; struct Matrix{ int mat[MAX][MAX]; Matrix operator +(const Matrix& t){ Matrix ans; for(int i=0;i<n;i++) for(int j=0;j<n;j++) ans.mat[i][j]=(this->mat[i][j]+ t.mat[i][j])%m; return ans; } Matrix operator *(const Matrix& t){ Matrix ans; memset(ans.mat,0,sizeof(ans.mat)); for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ for(int k=0;k<n;k++){ ans.mat[i][j]=(ans.mat[i][j]+ this->mat[i][k]*t.mat[k][j])%m; } } } return ans; } }; Matrix odd,I; void Pow(Matrix a,int x){ odd=I; while(x){ if(x&1) odd=odd*a; a=a*a; x>>=1; } } int main() { #ifndef ONLINE_JUDGE freopen("i.txt", "r", stdin); #endif int tk; l=0; Matrix S,a,t,P; cin>>n>>k>>m; for(int i=0;i<n;i++) I.mat[i][i]=1; for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ cin>>a.mat[i][j]; a.mat[i][j]%=m; } } memset(S.mat,0,sizeof(S.mat)); tk=k; while(tk){ vec[l++]=tk%2; tk>>=1; } S=a; tk=1; P=a; for(int i=l-2;i>=0;i--){ t=a; //Pow(t,tk); //S=S+odd*S; S=S+P*S; //tk=tk*2; P=P*P; if(vec[i]&1){ //++tk; P=P*a; //t=a; //Pow(t,tk); //S=S+odd; S=S+P; } } for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ if(j) cout<<" "; cout<<S.mat[i][j]; } cout<<endl; } return 0; }