基础练习 矩阵乘法(ACMORE1571)

基础练习 矩阵乘法(ACMORE1571)

题目

Description

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
例如：
A =
1 2
3 4
A的2次幂
7 10
15 22

Input

第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

Output

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

Sample Input

2 2
1 2
3 4

Sample Output

7 10
15 22

Source
蓝桥杯

题目链接: https://acmore.cc/problem/LOCAL/1571

简析

利用矩阵乘法法则即可，注意当m=0时，应输出单位矩阵。

AC的代码

Java版

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(new BufferedInputStream(System.in));
        int[][] t = new int[35][35];
        int[][] x = new int[35][35];
        int[][] r = new int[35][35];
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        if(m == 0)
            for(int i = 0; i < n; i++)
                for(int j = 0; j < n; j++)
                    if(i == j)
                        r[i][j] = 1;
                    else
                        r[i][j] = 0;
        else
            for(int i = 0; i < n; i++)
                for(int j = 0; j < n; j++)
                    x[i][j] = r[i][j] = scanner.nextInt();
        while(--m > 0){
            for(int i = 0; i < n; i++)
                for(int j = 0; j < n; j++)
                    for(int k = 0; k < n; k++)
                        t[i][j] += r[i][k] * x[k][j];
            for(int i = 0; i < n; i++)
                for(int j = 0; j < n; j++) {
                    r[i][j] = t[i][j];
                    t[i][j] = 0;
                }
        }
        for(int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++)
                System.out.print(r[i][j] + " ");
            System.out.println();
        }
    }
}