题目背景
l i n k link link
矩阵快速幂
题目描述
给定 n × n n×n n×n 的矩阵 A A A,求 A k A^k Ak
输入格式
第一行两个整数 n,k 接下来 n 行,每行 n 个整数,第 i 行的第 j 的数表示 A i , j A i,j Ai,j
输出格式
输出 A k A^k Ak
共 n 行,每行 n 个数,第 i 行第 j 个数表示 ( A k ) i , j (A^k)i,j (Ak)i,j ,每个元素对 1 0 9 + 7 10^9+7 109+7取模。
输入输出样例
输入 #1
2 1
1 1
1 1
输出 #1
1 1
1 1
分析:
矩阵乘法+矩阵快速幂模板 写法有很多种 注意 n , k n,k n,k开 l o n g long long l o n g long long
矩阵也要开 运算过程可能炸 i n t int int
也就是快速幂里用矩阵乘法 重载 ∗ * ∗号即可.
CODE:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=105;
const int mod=1000000007;
typedef long long ll;
ll n,k;
struct matrix{
ll G[N][N];
}A,B;
matrix operator *(matrix a,matrix b) //重载乘号
{
matrix C;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
C.G[i][j]=0;
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
C.G[i][j]=(C.G[i][j]+a.G[i][k]*b.G[k][j]%mod)%mod; //矩阵乘法
return C;
}
void ksm(ll x)
{
if(x==1){
B=A;
return;
}
ksm(x/2);
B=B*B; //快速幂
if(x&1) B=B*A;
}
int main(){
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%lld",&A.G[i][j]);
ksm(k);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
printf("%lld ",B.G[i][j]);
printf("\n");
}
return 0;
}