Tr A （矩阵快速幂）

Tr A

Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 108 Accepted Submission(s): 80

Problem Description A为一个方阵，则Tr A表示A的迹（就是主对角线上各项的和），现要求Tr(A^k)%9973。

Input 数据的第一行是一个T，表示有T组数据。 每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行，每行有n个数据，每个数据的范围是[0,9]，表示方阵A的内容。

Output             对应每组数据，输出Tr(A^k)%9973。

Sample Input 2 2 2 1 0 0 1 3 99999999 1 2 3 4 5 6 7 8 9

Sample Output 2 2686   就是矩阵快速幂， 求主对角线的和 啊。 #include <iostream> #include <cmath> #include <string> #include <algorithm> #include <string.h> #include <typeinfo> using namespace std; const int mod = 9973; int n; struct Matrix { int a[12][12]; }; Matrix operator * (const Matrix& a, const Matrix& b) { Matrix c; for (int i = 0; i < n; ++i) { for (int j = 0; j < n; ++j) { c.a[i][j] = 0; for (int k = 0; k < n; ++k) { c.a[i][j] += a.a[i][k] * b.a[k][j]; } c.a[i][j] %= mod; } } return c; } Matrix operator ^ (Matrix& a, int k) { Matrix c; for (int i = 0; i < n; ++i) for (int j = 0; j < n; ++j) { if (i == j) c.a[i][j] = 1; else c.a[i][j] = 0; } while (k) { if (k&1) c = a * c; k >>= 1; a = a * a; } return c; } int get_ans(const Matrix& m) { int ans = 0; for (int i = 0; i < n; ++i) { ans += m.a[i][i]; ans %= mod; } return ans; } int main() { int t; cin >> t; while (t--) { Matrix m; int k; cin >> n >> k; for (int i = 0; i < n; ++i) { for (int j = 0; j < n; ++j) { cin >> m.a[i][j]; } } m = m^k; cout << get_ans(m) << endl; } }