KY191 矩阵幂(用Java实现)

描述

给定一个n*n的矩阵,求该矩阵的k次幂,即P^k。

输入描述:

第一行:两个整数n(2<=n<=10)、k(1<=k<=5),两个数字之间用一个空格隔开,含义如上所示。 接下来有n行,每行n个正整数,其中,第i行第j个整数表示矩阵中第i行第j列的矩阵元素Pij且(0<=Pij<=10)。另外,数据保证最后结果不会超过10^8。

输出描述:

对于每组测试数据,输出其结果。格式为: n行n列个整数,每行数之间用空格隔开,注意,每行最后一个数后面不应该有多余的空格。

示例1

输入:

复制代码
2 2
9 8
9 3
3 3
4 8 4
9 3 0
3 5 7
5 2
4 0 3 0 1
0 0 5 8 5
8 9 8 5 3
9 6 1 7 8
7 2 5 7 3

输出:

复制代码
153 96
108 81
1216 1248 708
1089 927 504
1161 1151 739
47 29 41 22 16
147 103 73 116 94
162 108 153 168 126
163 67 112 158 122
152 93 93 111 97

代码如下:

java 复制代码
import java.util.Scanner;

/*
 * 矩阵幂
 */
public class MatrixPower {

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		while(scanner.hasNext()) {
			int n = scanner.nextInt();
			int k = scanner.nextInt();
			int[][] arr = new int[n][n];
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					arr[i][j] = scanner.nextInt();
				}
			}
			
			int[][] a = arr;
			
			for (int t = 0; t < k-1; t++) {
				
				//temp矩阵为arr矩阵的t+2次方
				int[][] temp = new int[n][n];
				
				//矩阵乘法的得到的结果保存到temp中
				for (int i = 0; i < n; i++) {
					for (int j = 0; j < n; j++) {
						for (int k1 = 0; k1 < n; k1++) {
							temp[i][j] += a[i][k1] * arr[k1][j];
						}
					}
				}
				
				a = temp;
			}
			
			for(int i = 0;i < n; i++) {
				for(int j = 0;j < n; j++) {
					System.out.print(a[i][j]+" ");
				}
				System.out.println();
			}
			
		}
		
	}
	
	
	
}
相关推荐
白白白小纯4 分钟前
每日算法day3—回文链表,链表分割
c语言·数据结构·算法·leetcode
编程风暴5 分钟前
3类人群的数据分析实习类型选择+4条选错红线
java·前端·数据分析
钱六两14 分钟前
Spring AI 使用 MCP 客户端(调用高德 MCP)
java·人工智能·spring
城管不管22 分钟前
RabbitMQ死信队列
java·分布式·ai·面试·职场和发展·rabbitmq·agent
手写码匠27 分钟前
华为云Flexus+DeepSeek征文|Dify 构建企业级联网搜索 Agent:查询改写、多源检索与引用溯源实战
人工智能·深度学习·算法·aigc
long31628 分钟前
Java 团队入门到精通学习资料
java·开发语言
七夜zippoe31 分钟前
DolphinDB 能耗统计分析实战:报表生成、同比环比与定额对比
人工智能·算法·dolphindb·报表生成·能耗统计·定额对比
vx-程序开发31 分钟前
【java项目分享】springboot农产品销售平台14952
java·javascript·spring boot·python·eclipse·django·php
为啥全要学41 分钟前
在大语言模型上使用 PPO 算法
人工智能·算法·语言模型
程序员良辰41 分钟前
【TongWeb7】启动接近两分钟问题排查
java·开发语言·中间件