问题描述
给定一个N阶矩阵A,输出A的M次幂(M是非负整数)
例如:
A =
1 2
3 4
A的2次幂
7 10
15 22
例如:
A =
1 2
3 4
A的2次幂
7 10
15 22
输入格式
第一行是一个正整数N、M(1<=N<=30, 0<=M<=5),表示矩阵A的阶数和要求的幂数
接下来N行,每行N个绝对值不超过10的非负整数,描述矩阵A的值
接下来N行,每行N个绝对值不超过10的非负整数,描述矩阵A的值
输出格式
输出共N行,每行N个整数,表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开
样例输入
2 2
1 2
3 4
1 2
3 4
样例输出
7 10
15 22
15 22
题目分析:题意明确,矩阵乘法
算法分析:反思一下,矩阵乘法都会乘,但是通用公式呢,能信手拈来吗,思路不清晰的尽早百度百科好好复习为上策。否则光一个乘法公式都需要慢慢推导是真的费时费力了。另外,题目中还有一个注意点,就是M是非负整数,矩阵的零次幂是单位阵,不能忘!
算法设计:
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int N = sc.nextInt();
int M = sc.nextInt();
long[][] a = new long[N][N];
long[][] b = new long[N][N];
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
b[i][j] = a[i][j] = sc.nextLong();
}
}
sc.close();
// 矩阵的0次幂为单位阵
if (M == 0) {
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (i == j)
System.out.print(1 + " ");
else
System.out.print(0 + " ");
}
System.out.println();
}
} else if (M == 1) { // 矩阵的1次幂为本身
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
System.out.print(a[i][j] + " ");
System.out.println();
}
}
} else {
for (int x = 1; x < M; x++) {
long[][] temp = new long[N][N];
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
long result = 0;
for (int k = 0; k < N; k++) {
result += a[i][k] * b[k][j]; // 矩阵乘法
}
temp[i][j] = result;
}
}
b = temp;
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
System.out.print(b[i][j] + " ");
}
System.out.println();
}
}
}
}