poj 3233 Matrix Power Series
生活随笔
收集整理的這篇文章主要介紹了
poj 3233 Matrix Power Series
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
最近一直在做矩陣的題目,這算是一道簡單題目吧。
建立一個矩陣的結構體,便于賦值和返回值。
S?=?A?+?A2?+?A3?+ … +?Ak
? ??如果 S=A+A2?+?A3?+A4 ? S= (A?+?A2)*(E+A2) ?所以只需要 二分操作就可以了
? ?再輔助以 快速冪,矩陣乘法,矩陣加法就可以了
#include<cstdio> #include<cstring> #include<iostream> using namespace std; struct matrix{int f[40][40]; }; int n,m; matrix pow(matrix a,matrix b) {int i,j,k;matrix c;memset(c.f,0,sizeof(c.f));for(i=0;i<n;i++)for(j=0;j<n;j++)for(k=0;k<n;k++)c.f[i][j]=(c.f[i][j]+a.f[i][k]*b.f[k][j])%m;return c; } matrix quick_pow(matrix a,int k) {matrix s;memset(s.f,0,sizeof(s.f));int i;for(i=0;i<n;i++) s.f[i][i]=1;while(k){if(k&1)s=pow(s,a);a=pow(a,a);k=k>>1;}return s; } matrix add(matrix a,matrix b) {int i,j;for(i=0;i<n;i++)for(j=0;j<n;j++)a.f[i][j]=(a.f[i][j]+b.f[i][j])%m;return a; } matrix solve(matrix A,int k) {matrix E;memset(E.f,0,sizeof(E.f));int i;for(i=0;i<n;i++) E.f[i][i]=1;if(k==1) return A;else if(k&1)return add(solve(A,k-1),quick_pow(A,k));elsereturn pow(solve(A,k/2),(add(quick_pow(A,k/2),E))); } int main() {int i,j,k;matrix prematrix;scanf("%d%d%d",&n,&k,&m);for(i=0;i<n;i++)for(j=0;j<n;j++)scanf("%d",&prematrix.f[i][j]);matrix ans=solve(prematrix,k);for(i=0;i<n;i++){for(j=0;j<n;j++){if(j==0) printf("%d",ans.f[i][j]%m);else printf(" %d",ans.f[i][j]%m);}printf("\n");} }超強干貨來襲 云風專訪:近40年碼齡,通宵達旦的技術人生
總結
以上是生活随笔為你收集整理的poj 3233 Matrix Power Series的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: POJ 3311 Hie with th
- 下一篇: POj 3420 Quad Tiling