59-题目1443：Tr A

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;
}