http://ac.jobdu.com/problem.php?pid=1443
-
题目描述:
-
A为一个方阵,则Tr A表示A的迹(就是主对角线上各项的和),现要求Tr(A^k)%9973。
-
输入:
-
数据的第一行是一个T,表示有T组数据。
每组数据的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)两个数据。接下来有n行,每行有n个数据,每个数据的范围是[0,9],表示方阵A的内容。
-
输出:
-
对应每组数据,输出Tr(A^k)%9973。
本来我以为可以用一维数组代替矩阵的二维数组,但是我想错了。。。我把A^k当成直接让对角线元素^k了。。。。真是低级!
动态建立二维数组的代码:(虽然没用上,因为要用的矩阵比较多,不能每一个都建立二维数组,这样太麻烦了)
int **A = new int*[n];
for (int i = 0; i < n; i++)
A[i] = new int[n];
花了好长时间啊。。。。结果是一个很低级的错误。。。把%写成了/~~~~~~找了一上午这个错误。。。
#include<iostream>
#include<fstream>
#include<math.h>
using namespace std;
#define M 9973
struct Matrix
{
int n[10][10];
};
Matrix E; //单位矩阵
int n; //n作为全局变量,表示矩阵的规模
Matrix multiple(Matrix A, Matrix B) //矩阵相乘
{
int i, j, k;
Matrix temp;
for (i = 0; i < n; i++) //矩阵相乘
for (j = 0; j < n; j++)
{
temp.n[i][j] = 0;
for (k = 0; k < n; k++)
{
temp.n[i][j] += (A.n[i][k] * B.n[k][j]) % M;
temp.n[i][j] %= M;
}
}
return temp;
}
Matrix divide_power(Matrix A, int k)
{
Matrix ans = E;
while (k != 0)
{
if (k % 2 == 1) //k为奇数时
ans = multiple(ans, A); //ans = ans*A ,虽然最终结果只是要对角线上元素,但是矩阵相乘时其他元素还是要考虑
k /= 2;
A = multiple(A, A);
}
return ans;
}
int main()
{
int T, k, i, j, sum;
Matrix A, temp;
ifstream cin("data.txt");
for (i = 0; i < 10; i++) //单位矩阵
for (j = 0; j < 10; j++)
{
if (i == j)
E.n[i][j] = 1;
else
E.n[i][j] = 0;
}
cin >> T;
while (T--)
{
cin >> n >> k;
for (i = 0; i < n; i++) //输入矩阵数据
for (j = 0; j < n; j++)
cin >> A.n[i][j];
temp = divide_power(A, k);
sum = 0;
for (i = 0; i < n; i++) //求tr
sum += temp.n[i][i] % M;
sum %= M;
cout << sum << endl;
}
system("pause");
return 0;
}