Tr A hdu1575

Tr A

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1781    Accepted Submission(s): 1311

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
  
  

 

 

 

import java.util.Scanner;

public class HDU1575 {

	public static int[][] arr;
	public static int n, k;
	public static int mod = 9973;
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int cases = sc.nextInt();
		while(cases-- > 0) {
			n = sc.nextInt();
			k = sc.nextInt();
			arr = new int[n][n];
			for(int i=0; i<n; i++) {
				for(int j=0; j<n; j++) {
					arr[i][j] = sc.nextInt();
				}
			}
			sovle();
		}
	}
	
	private static void sovle() {
		int[][] result = new int[n][n];
		for(int i=0; i<n; i++) {//初始为单位矩阵
			result[i][i] = 1;
		}
		while(k>0) {
			if((k&1) == 1) {
				result = matrixMul(result, arr);
			}
			arr = matrixMul(arr, arr);
			k >>= 1;
		}
		int r = 0;
		for(int i=0; i<n; i++) {
			r = (r+result[i][i])%mod;
		}
		System.out.println(r);
	}
	
	public static int[][] matrixMul(int[][] a1, int[][] a2) {
		int[][] result = new int[n][n];
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				for(int k=0; k<n; k++) {
					result[i][j] = (result[i][j]+a1[i][k]*a2[k][j])%mod;
				}
			}
		}
		return result;
	}

}