java：蓝桥Basic_17：矩阵乘法

蓝桥Basic_17

问题描述

• 给定一个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 Basic_17 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n,m;
        n = scanner.nextInt();
        m = scanner.nextInt();
        int[][] matrix = new int[n][n];
        //录入矩阵
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                matrix[i][j] = scanner.nextInt();
            }
        }
        //矩阵matrix计算
        if(m==0){
            //当m等于0时，为E
            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 {
            //当m不等于0时，矩阵计算
            int[][] matrix1 = matrix;
            /**易错点：此处i不是等于0，而是从1开始，因为第一次就是两个矩阵开始计算*/
            for (int i = 1; i < m; i++) {//m是多少就是多少个矩阵计算
                matrix1 = multiply(matrix,matrix1);
            }
            print(matrix1);
        }
    }

    private static void print(int[][] matrix1) {
        for (int i = 0; i < matrix1.length; i++) {
            for (int j = 0; j < matrix1.length; j++) {
                System.out.print(matrix1[i][j]+" ");
            }
            System.out.println();
        }
    }

    private static int[][] multiply(int[][] matrix, int[][] matrix1) {
        int[][] ans = new int[matrix.length][matrix.length];
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix.length; j++) {
                for (int k = 0; k < matrix.length; k++) {
                    /**
                     * 先找出固定的值，再找出变化的值
                     * 每一个行乘列的时候，行的x(i)左边不变，列的y(j)左边不变,k时刻变化
                     * */
                    ans[i][j] += matrix[i][k]*matrix1[k][j];
                }
            }
        }
        return ans;
    }
}