44 lines
1.2 KiB
C++
44 lines
1.2 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
#define int long long
|
||
#define endl "\n"
|
||
const int MOD = 1e9 + 7;
|
||
const int N = 110;
|
||
int n, k; // 本题数据范围很大，用int直接wa哭了
|
||
int a[N][N], b[N][N]; // 原始矩阵，结果矩阵
|
||
|
||
// 矩阵乘法
|
||
void mul(int a[][N], int b[][N], int c[][N]) {
|
||
int t[N][N] = {0};
|
||
for (int i = 0; i < N; i++)
|
||
for (int j = 0; j < N; j++)
|
||
for (int k = 0; k < N; k++)
|
||
t[i][j] = (t[i][j] + a[i][k] * b[k][j] % MOD) % MOD; // 矩阵乘法
|
||
memcpy(c, t, sizeof t);
|
||
}
|
||
|
||
int main() {
|
||
cin >> n >> k;
|
||
// 1、输入原始矩阵
|
||
for (int i = 0; i < n; i++)
|
||
for (int j = 0; j < n; j++) {
|
||
cin >> a[i][j];
|
||
b[i][j] = a[i][j]; // 这种解法与标准解法不同，不用构建单位矩阵
|
||
// 直接赋值初始化了一个，然后执行k-1次就完成了矩阵的k次幂
|
||
}
|
||
|
||
// 2、计算矩阵快速幂
|
||
for (int i = k - 1; i; i >>= 1) {
|
||
if (i & 1) mul(a, b, b);
|
||
mul(a, a, a);
|
||
}
|
||
|
||
// 3、输出
|
||
for (int i = 0; i < n; i++) {
|
||
for (int j = 0; j < n; j++)
|
||
printf("%lld ", b[i][j]);
|
||
printf("\n");
|
||
}
|
||
|
||
return 0;
|
||
} |