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();
			}
			
		}
		
	}
	
	
	
}
相关推荐
lingran__18 小时前
速通ACM省铜第三天 赋源码(Double Perspective和Trip Shopping和Hamiiid, Haaamid... Hamid?)
c++·算法
凤城老人18 小时前
C++使用拉玛努金公式计算π的值
开发语言·c++·算法
失散1319 小时前
分布式专题——10.4 ShardingSphere-Proxy服务端分库分表
java·分布式·架构·shardingsphere·分库分表
时空无限20 小时前
说说transformer 中的掩码矩阵以及为什么能掩盖住词语
人工智能·矩阵·transformer
Bellafu66620 小时前
spring项目部署后为什么会生成 logback-spring.xml文件
java
递归不收敛20 小时前
一、Java 基础入门:从 0 到 1 认识 Java(详细笔记)
java·开发语言·笔记
沐浴露z21 小时前
【Java SpringAI智能体开发学习 | 2】SpringAI 实用特性:自定义Advisor,结构化输出,对话记忆持久化,prompt模板,多模态
java·spring·springai
小沈同学呀21 小时前
创建一个Spring Boot Starter风格的Basic认证SDK
java·spring boot·后端
纪元A梦21 小时前
贪心算法应用:配送路径优化问题详解
算法·贪心算法
码农小伙21 小时前
通俗易懂地讲解JAVA的BIO、NIO、AIO
java·nio