社区讨论

矩阵快速幂板子求调qwq

灌水区参与者 3已保存回复 2

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
2 条
当前快照
1 份
快照标识符
@lvnww940
此快照首次捕获于
2024/05/01 22:27
2 年前
此快照最后确认于
2024/05/02 10:55
2 年前
查看原帖
rt,一个点 MLE。
CPP
#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;
long long n,k;
struct Matrix{
	long long Mat[105][105];
}tmp,a;
Matrix operator *(Matrix a,Matrix b){
	Matrix c;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			for(int k=1;k<=n;k++){
				c.Mat[i][j]+=a.Mat[i][k]*b.Mat[k][j];
				c.Mat[i][j]%=mod;
			}
		}
	}
	return c;
}
Matrix tmplol;
inline Matrix ksm(Matrix a,long long k){
    if(k==1){
        return a;
    }
	if(k%2==0){
		tmplol=ksm(a,k/2);
		return tmplol*tmplol;
	}
	else{
        tmplol=ksm(a,k/2);
        return tmplol*tmplol*a;
	}
}
int main(){
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>a.Mat[i][j];
		}
	}
	tmp=ksm(a,k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<tmp.Mat[i][j]<<" ";
		}
		cout<<endl;
	}
	return 0;
}

回复

2 条回复,欢迎继续交流。

正在加载回复...