矩阵快速幂（模板）

P3390 

#include<iostream>
#include<cstring>
#define ll long long
using namespace std;
const int N=105;
const int mod = 1e9+7;
int n;
struct Matrix{
	ll matrix[N][N];
};
void init(Matrix &res) {	//初始化为单位矩阵 
	memset(res.matrix,0,sizeof(res.matrix));
	for(int i=1;i<=N;i++)
		res.matrix[i][i] = 1;
}
Matrix multiply(Matrix a,Matrix b) {  //矩阵乘法 
	Matrix res;
	memset(res.matrix,0,sizeof(res.matrix));
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=n;j++) {
			for(int k=1;k<=n;k++) {
				res.matrix[i][j] += a.matrix[i][k]*b.matrix[k][j];
				res.matrix[i][j] %= mod;
			}
		}
	}
	return res;
}
Matrix Pow(Matrix x,ll m) {  //矩阵快速幂 x^m   m:指数   n:矩阵大小 
	Matrix res;
	init(res);
	while(m) {
		if(m&1)
			res = multiply(res,x);
		x = multiply(x,x);
		m>>=1; 
	}
	return res;
} 
int main() {
	ll m;
	cin >> n >> m;
	Matrix a,b; 
	for(int i=1;i<=n;i++) 
		for(int j=1;j<=n;j++)
			cin >> a.matrix[i][j];
	b = Pow(a,m);
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=n;j++)
			cout << b.matrix[i][j] << " ";
		cout<<endl;
	}
	return 0;
}

 

可用于矩阵加速。