HDU 1575 Tr A(矩阵快速幂)

最新推荐文章于 2020-01-18 16:14:59 发布

小鱼爱吃火锅

最新推荐文章于 2020-01-18 16:14:59 发布

阅读量90

点赞数

本文链接：https://blog.csdn.net/daoshen1314/article/details/102576723

版权

矩阵快速幂专栏收录该内容

11 篇文章 0 订阅

订阅专栏

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

Sample Output

2
2686

模板题：

import java.util.Arrays;
import java.util.Scanner;

public class main18 {
	static int n,k;
	static final int max=15;
	static final int mod=9973;
	public static int[][] multi(int a[][],int b[][]){
		int res[][]=new int[max][max];
		for(int i=0;i<n;i++)
			Arrays.fill(res[i], 0);
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				for(int k=0;k<n;k++)
					res[i][j]=(res[i][j]+a[i][k]*b[k][j]%mod)%mod;
		return res;
	}
	public static int[][] quick_pow(int a[][]){
		int res[][]=new int[max][max];
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				if(i==j) res[i][j]=1;
				else res[i][j]=0;
		int b=k;
		while(b!=0){
			if((b&1)==1) res=multi(res,a);
			b/=2;
			a=multi(a,a);
		}
		return res;
	}
    public static void main(String[] args) {
		Scanner scan=new Scanner(System.in);
		int t=scan.nextInt();
		while(t>0){
			n=scan.nextInt();
			k=scan.nextInt();
			int a[][]=new int[max][max];
			for(int i=0;i<n;i++)
				for(int j=0;j<n;j++)
					a[i][j]=scan.nextInt();
			int b[][]=new int[max][max];
			b=quick_pow(a);
			int ans=0;
			for(int i=0;i<n;i++){
				ans=(ans+b[i][i]%mod)%mod;
			}
			System.out.println(ans);
			t--;
		}
	}
}