玄关,优化10pts代码
查看原帖
玄关,优化10pts代码
560291
the_Dream_Chaser楼主2024/5/10 20:06

10pts 代码求调,全TLE了,不知道如何提速

#include<cstdio>
using namespace std;
#define ll long long
const int HBG=1e9+7;
const int MAXN=105;
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;k++){
				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;k++){
				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:06
加载中...