题意:给你一个矩阵A,求S=A+A^2+A^3+...+A^k。
其实这个当时我看着毫无头绪,看了他们给的矩阵发现好!精!妙!
我们这样看
是不是有点思路!
没错!就是右上角,我们以此类推可以得到A+A^2+A^3+...+A^k+E,我们只要再减去个单位矩阵就好了。
但是!我矩阵里面怎么套矩阵!肿!么!办!其实很简单,一个n*n的矩阵,我们可以把它看成(2*n)*(2*n)的矩阵,就把他分成了四份,就如上图所示,就很简单了!
注意下小坑点:减了可能就负了,后面减完要加个mod(ง •_•)ง
#include<cstdio> #include<cmath> #include<iostream> #include<algorithm> #include<vector> #include<stack> #include<cstring> #include<queue> #include<set> #include<string> #include<map> #include <time.h> #define PI acos(-1) using namespace std; typedef long long ll; typedef double db; const int maxn = 62; const int N = 10; const ll maxm = 1e7; const int INF = 0x3f3f3f; const ll inf = 1e15 + 5; const db eps = 1e-9; ll n, k, mod; struct Matrix{ ll mat[maxn][maxn]; Matrix operator*(const Matrix& m)const{ Matrix tmp; for (int i = 1; i <= n; i++) { for (int j = 1; j <= n; j++) { tmp.mat[i][j]=0; for (int k = 1; k <= n; k++) { tmp.mat[i][j] += mat[i][k]*m.mat[k][j]%mod; tmp.mat[i][j]+=mod; tmp.mat[i][j] %= mod; } } } return tmp; } }; int Pow(Matrix &m, int k) { Matrix ans; memset(ans.mat , 0 , sizeof(ans.mat)); for (int i=1; i<=n; i++) { ans.mat[i][i]=1; ans.mat[i+n][i+n]=1; } n*=2; while(k){ if(k&1) ans = ans*m; k >>= 1; m = m*m; } n/=2; for (int i=1; i<=n; i++) { ans.mat[i][i+n]--; ans.mat[i][i+n]+=mod; ans.mat[i][i+n]%=mod; } for (int i=1; i<=n; i++) { for (int j=1; j<=n; j++) { if (j==n) printf("%d ", ans.mat[i][j+n]); else printf("%d ", ans.mat[i][j+n]); } } } void solve() { Matrix m; memset(m.mat, 0, sizeof(m.mat)); scanf("%lld%lld%lld", &n, &k, &mod); for (int i = 1; i <= n; i++) { for (int j = 1; j <= n; j++) { scanf("%lld", &m.mat[i][j]); m.mat[i][i+n]=1; m.mat[i+n][i+n]=1; } } k++; Pow(m, k); } int main() { int t = 1; //freopen("in.txt", "r", stdin); // scanf("%d", &t); while(t--) solve(); return 0; }