蓝桥杯 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

问题分析

方阵的乘法某一个值得计算可表示为：
res[i][j] = sum(a[i][k] * b[k][j])，(k∈[1,n],n为阶数)。
这道题里不过是两个相同的矩阵相乘。同时要特殊考虑次幂为0的情况（单位矩阵），由于二维矩阵本身的计算就有两层for占据了大量的篇幅，方便起见抽取为函数表示。

代码

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner read = new Scanner(System.in);
        int N = read.nextInt();
        int M = read.nextInt();
        int [][]a = new int[N][N];
        int [][]b = new int[N][N];
        b = a;

        //输入矩阵
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                a[i][j] = read.nextInt();
            }
        }

        //幂运算
        if (M==0) b = calculate(N,a,b,M);
        else {
            for(int m=1;m<M;m++){
                b = calculate(N,a,b,M);
            }
        }

        //打印
        print(N,b);


    }


    //打印
    private static void print(int N,int[][] b){
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                System.out.printf(b[i][j]+" ");
            }
            System.out.println();
        }
    }



    //计算
    private static int[][] calculate(int N, int[][] a, int[][] b,int M) {
        int[][] res = new int[N][N];
        if(M == 0){
            for(int i=0;i<N;i++){
                res[i][i] = 1;
            }
        }
        else{
            for(int i = 0; i< N; i++){
                for(int j = 0; j< N; j++){
                    for(int k = 0; k< N; k++){
                        res[i][j] += a[i][k] * b[k][j];
                    }
                }
            }
        }

        return res;
    }

}