矩阵快速幂例3：矩阵幂级数

最新推荐文章于 2024-06-06 16:05:02 发布

原创最新推荐文章于 2024-06-06 16:05:02 发布 · 1.8k 阅读

CC 4.0 BY-SA版权

这篇博客介绍了矩阵快速幂算法的实现，并通过一个具体的例3：矩阵幂级数来详细解析该算法的过程。博主首先定义了矩阵乘法函数`mu`，然后实现了矩阵快速幂的`answer`函数，用于计算矩阵的幂次。在`main`函数中，读取输入的矩阵和幂次，最后输出计算结果。矩阵快速幂算法在处理大型矩阵运算时能显著提高效率。

部署运行你感兴趣的模型镜像

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n,k;
int a[4005][4005],b[4005][4005];int c[4005][4005],t[4005][4005];
void mu(int a[][4005],int b[][4005],int n){
memset(c,0,sizeof c);
for(int i=1;i<=2*n;i++){
for(int j=1;j<=2*n;j++){
for(int k=1;k<=2*n;k++){
c[i][j]+=a[i][k]*b[k][j];
c[i][j]%=1000;
}
}
}
for(int i=1;i<=2*n;i++){
for(int j=1;j<=2*n;j++){
a[i][j]=c[i][j];
//printf("%d ",a[i][j]);
}
//printf("\n");
}
return;
}
void answer(int a[][4005],int k){
while(k){
if(n&1) mu(a,b,n);
mu(b,b,n);
k/=2;
}
}
int main(){
freopen("7.in","r",stdin);
freopen("7.out","w",stdout);
scanf("%d%d",&n,&k);
memset(a,0,sizeof a);
memset(b,0,sizeof b);
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
scanf("%d",&a[i][j]);
b[i][j]=a[i][j];
b[n+i][j]=a[i][j];
}
}
//printf("%d\n",b[1][1]);
for(int i=n+1;i<=2*n;i++){
for(int j=n+1;j<=2*n;j++){
b[i][j]=1;
}
}
answer(a,k);
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
printf("%d ",a[i][j]);
}
printf("\n");
}
return 0;
}

您可能感兴趣的与本文相关的镜像