【数论】矩阵快速幂

模板如下:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=110;
const int mod=1e9+7;
#define mod(x) ((x)%mod)

int n;
struct mat{
	int m[maxn][maxn];
}unit;

mat operator * (mat a,mat b){
	mat ret;
	ll x;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			x=0;
			for(int k=0;k<n;k++)
				x+=mod((ll)a.m[i][k]*b.m[k][j]);
			ret.m[i][j]=mod(x);	
		}
	}
	return ret;
} 

void init_unit(){
	for(int i=0;i<maxn;i++){
		unit.m[i][i]=1;
	}
	return ;
}

mat pow_mat(mat a,ll n){
	mat ret=unit;
	while(n){
		if(n&1) ret=ret*a;
		a=a*a;
		n>>=1;
	}
	return ret;
}

int main(){
	ll x;
	init_unit();
	while(cin>>n>>x){
		mat a;
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				cin>>a.m[i][j];
			}
		}
		a=pow_mat(a,x);
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				if(j+1==n) cout<<a.m[i][j]<<endl;
				else cout<<a.m[i][j]<<" ";
			}
		}
	}
	return 0;
}
发布了71 篇原创文章 · 获赞 5 · 访问量 3396

猜你喜欢

转载自blog.csdn.net/Rainfoo/article/details/103996538