标签：hdu1575

Tr A

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

2
2686

A,B两矩阵相乘得到的新矩阵的第i行j列值为A矩阵的第i行每个元素跟B矩阵第j列每个元素相乘的和。给矩阵赋值为1的话,需要把主对角线赋值成1，其余部分为0.

#include <stdio.h>
#include <string.h>

#define mod 9973
typedef __int64 LL;

struct Node {
	LL mat[12][12];
} A;
int n, k;

Node mat_multi(Node A, Node B) {
	int i, j, k;
	Node c;
	memset(c.mat, 0, sizeof(c.mat));
	for(i = 0; i < n; ++i)
		for(j = 0; j < n; ++j) {
			for(k = 0; k < n; ++k) {
				c.mat[i][j] += A.mat[i][k] * B.mat[k][j];
				c.mat[i][j] %= mod;
			}
		}
	return c;
}

Node power_mod(Node A, int k) {
	Node c;
	memset(c.mat, 0, sizeof(c.mat));
	int i, j;
	for(i = 0; i < n; ++i)
		c.mat[i][i] = 1;

	for( ; k > 0; k >>= 1) {
		if(k & 1) c = mat_multi(c, A);
		A = mat_multi(A, A);
	}
	return c;
}

int main() {
	int T, i, j, ans;
	scanf("%d", &T);
	while(T--) {
		scanf("%d%d", &n, &k);
		for(i = 0; i < n; ++i)
			for(j = 0; j < n; ++j)
				scanf("%I64d", &A.mat[i][j]);
		A = power_mod(A, k);
		ans = 0;
		for(i = 0; i < n; ++i)
			ans += A.mat[i][i];
		printf("%d\n", ans % mod);
	}
	return 0;
}

HDU1575 Tr A 【矩阵快速幂】

标签：hdu1575

原文地址：http://blog.csdn.net/chang_mu/article/details/41412299