洛谷 P3390 【模板】矩阵快速幂

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
long long num[105][105],l,ji[105][105],ji2[105][105];
int n,i,j,k;
void jzcf(long long a[105][105],long long b[105][105]) //由于这里传入的是地址因此可以赋值。
{
for(i = 1;i <= n;i ++)
for(j = 1;j <= n;j ++)
ji2[i][j] = a[i][j],a[i][j] = 0; //用a来储存答案。
if(a == b) //若a与b的地址相同 ，则把b的地址修改数值相同的ji。
b = ji2;
for(i = 1;i <= n;i ++)
for(j = 1;j <= n;j ++)
for(k = 1;k <= n;k ++)
a[i][j] = (a[i][j] + (b[i][k] * ji2[k][j])) % 1000000007;
// 第i行的每一个数 * 第j列的每一个数之和就是a[i][j]的值。
}
void sqr(long long x) //快速幂。
{
while(x)
{
if(x & 1 == 1)
jzcf(ji,num);
jzcf(num,num);
x = x >> 1;
}
}
int main()
{
scanf("%d%lld",&n,&l);
for(i = 1;i <= n;i ++)
for(j = 1;j <= n;j ++)
scanf("%lld",&num[i][j]),ji[i][j] = num[i][j]; //ji是原矩阵 * 1；
sqr(l - 1); // 已经有了一个 * 1的矩阵。
for(i = 1;i <= n;i ++)
{
for(j = 1;j <= n;j ++)
printf("%lld ",ji[i][j]);
puts("");
}
}