10pts,全TLE,玄关
查看原帖
10pts,全TLE,玄关
560291
the_Dream_Chaser楼主2024/5/10 20:20

大佬求调,10pts,全为TLE,不知如何提速

#include<cstdio>
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int HBG=1e9+7;
const int MAXN=116;
ll a[MAXN][MAXN],ans[MAXN][MAXN]={0};
ll n,m;
//单位矩阵:与单位矩阵相乘,还是原矩阵 
/*	1 0 0 ...  
	0 1 0 ... 
	0 0 1 ... 
	...
*/
inline ll read(){
	char c=getchar();
	ll f=1,x=0;
	while(c<'0'||c>'9'){
		if(c=='-')
			f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		x=(x<<1)+(x<<3)+(c^'0');
		c=getchar();
	}
	return x*f;
}
inline void HHZ(){//ans=ans*a
	ll c[MAXN][MAXN]={0};
	for(int k=1;k<=n;k++){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				c[i][j]=(c[i][j]+ans[i][k]*a[k][j])%HBG;
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			ans[i][j]=c[i][j];
		}
	}
}
inline void XMX(){//a=a*a
	ll c[MAXN][MAXN]={0};
	for(int k=1;k<=n;k++){
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				c[i][j]=(c[i][j]+a[i][k]*a[k][j])%HBG;
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			a[i][j]=c[i][j];
		}
	}
}

int main(){
	n=read();m=read();
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){//输入 
			a[i][j]=read();
		}
	}
	for(int i=1;i<=n;i++)ans[i][i]=1;//设置成单位矩阵
	//快速幂
	
	while(m){
		if(m&1){
			
			HHZ();//ans*=a
			XMX();//a*=a
			m>>=1;
		}
	} 
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			printf("%lld ",ans[i][j]%HBG);
		}
		printf("\n");
	} 
}
2024/5/10 20:20
加载中...