蓝桥杯试题 基础练习 矩阵乘法 BASIC-17 JAVA

前言

最近一直搞面试，很多写好的代码都懒得去发博客，现在补上，但是注释可能比较少，大家如果有问题请联系我

试题 基础练习 矩阵乘法

资源限制
时间限制：1.0s 内存限制：512.0MB

问题描述
　　给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　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个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

样例输入
2 2
1 2
3 4

样例输出
7 10
15 22

本题代码

import java.util.Scanner;

public class MatrixMulti {

    private static int N;//矩阵A的阶数
    private static int[][] matrix; //矩阵A

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        int M = sc.nextInt(); //要求的幂数
        matrix = new int[N][N];
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                matrix[i][j] = sc.nextInt();
            }
        }
        int[][] newMatrix = new int[N][N];
        if (M == 0) {
            for (int i = 0; i < N; i++) {
                newMatrix[i][i] = 1;
            }
        } else {
            newMatrix = matrix;
            for (int i = 0; i < M - 1; i++) {
                newMatrix = matriMultiplication(newMatrix);
            }
        }

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                System.out.print(newMatrix[i][j] + " ");
            }
            System.out.println();
        }
    }

    private static int[][] matriMultiplication(int[][] matrixParam) {
        int[][] newMatrix = new int[N][N];
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                for (int k = 0; k < N; k++) {
                    newMatrix[i][k] += matrixParam[i][j] * matrix[j][k];
                }
            }
        }
        return newMatrix;
    }
}