矩阵快速幂

题目描述

给定 n×n 的矩阵 A，求 A^k。

输入格式

第一行两个整数 n,k 接下来 n 行，每行 n 个整数，第 i 行的第 j 的数表示 A_{i,j}。

输出格式

输出 A^k

共 nn 行，每行 nn 个数，第 ii 行第 jj 个数表示 (A^k)_{i,j}(Ak)i,j​，每个元素对 10^9+7109+7 取模。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=200;
ll n,k;
struct name{
	ll g[N][N];
};
struct name los;
name operator*(name x,name y){
	struct name a;
	memset(&a,0,sizeof a);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int k=1;k<=n;k++){
				a.g [i][j]=((a.g [i][j]+x.g [i][k]*y.g [k][j])%mod)%mod;
			}
		}
	}
	return a;
}
void ksm(ll k){
	struct name res;
	memset(&res,0,sizeof res);
	for(int i=1;i<=n;i++){
		res.g [i][i]=1;
	}
	while(k){
		if(k&1)res=res*los;
		los=los*los;
		k>>=1;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			printf("%lld ",res.g [i][j]);
		}
		printf("\n");
	}
}
int main(){
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>los.g [i][j];
		}
	}
	ksm(k);
	return 0;
}