WZOI-584矩阵快速幂

题目描述:

给定n*n的矩阵A，求A^k

输入格式:

第一行，n,k

第2至n+1行，每行n个数，第i+1行第j个数表示矩阵第i行第j列的元素

输出格式:

输出A^k

共n行，每行n个数，第i行第j个数表示矩阵第i行第j列的元素，每个元素模10^9+7

样例输入:

2 1
1 1
1 1

样例输出:

1 1

1 1
 

代码如下：

#include
using namespace std;
long long n,a[101][101]={};
long long s[101][101]={};
int const x=1e9+7;
void ju(long long b[101][101],long long c[101][101]){
	long long d[101][101]={};
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int ii=1;ii<=n;ii++){
				d[i][j]+=(b[i][ii]*c[ii][j])%x;	
				d[i][j]=d[i][j]%x;	
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			b[i][j]=d[i][j];
		}
	}
}
void kuai(long long k){
	
	for(int i=1;i<=n;i++){
		s[i][i]=1;
	}
	while(k){
		if(k&1){
			ju(s,a);
		}
		ju(a,a);
		k/=2;
	}
}
int main(){
	long long k;
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>a[i][j];
		}
	}
	kuai(k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<" ";
		}cout<
	}
	return 0;
}