'commit'

TangDou/AcWing_TiGao/T5/JuZhengChengFa/P3390_2.cpp

@@ -0,0 +1,44 @@
+#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;
+}