题目背景
矩阵快速幂
题目描述
给定n*n的矩阵A,求A^k
输入输出格式
输入格式:
第一行,n,k
第2至n+1行,每行n个数,第i+1行第j个数表示矩阵第i行第j列的元素
输出格式:
输出A^k
共n行,每行n个数,第i行第j个数表示矩阵第i行第j列的元素,每个元素模10^9+7
输入输出样例
输入样例#1:
2 1
1 1
1 1
输出样例#1:
1 1
1 1
说明
矩阵快速幂
题目描述
给定n*n的矩阵A,求A^k
输入输出格式
输入格式:
第一行,n,k
第2至n+1行,每行n个数,第i+1行第j个数表示矩阵第i行第j列的元素
输出格式:
输出A^k
共n行,每行n个数,第i行第j个数表示矩阵第i行第j列的元素,每个元素模10^9+7
输入输出样例
输入样例#1:
2 1
1 1
1 1
输出样例#1:
1 1
1 1
说明
n<=100, k<=10^12, |矩阵元素|<=1000 算法:矩阵快速幂
吐槽:wa了两次太丢人了,第一次发现没有取模,第二次发现没有开long long,简直了,QAQ。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define mod 1000000007
using namespace std;
int n;
long long m;
struct node
{
int cal,line;
long long a[100][100];
};
node pow(node x,node y)
{
node re;
memset(re.a,0,sizeof(re.a));
re.cal=x.cal;re.line=y.line;
for(int i=1;i<=x.cal;i++)
for(int k=1;k<=y.line;k++)
for(int j=1;j<=x.line;j++)
re.a[i][k]=(re.a[i][k]+(x.a[i][j]*y.a[j][k])%mod)%mod;
return re;
}
int main()
{
node tmp,ans;
scanf("%d%lld",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%lld",&tmp.a[i][j]);
for(int i=1;i<=n;i++)
ans.a[i][i]=1;
ans.cal=n;ans.line=n;tmp.cal=n;tmp.line=n;
while(m)
{
if(m&1) ans=pow(ans,tmp);
tmp=pow(tmp,tmp);
m>>=1;
}
for(int i=1;i<=ans.cal;i++)
{
for(int j=1;j<ans.line;j++)
printf("%lld ",ans.a[i][j]);
printf("%lld\n",ans.a[i][ans.line]);
}
return 0;
}