蓝桥杯 基础练习VIP 矩阵乘法 java

题目

题目描述
给定一个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

代码

矩阵的零次幂 为 单位矩阵。
2 2
2 2 的 0 次幂为

1 0
0 1

---

import java.io.IOException;
import java.util.Scanner;

public class Main {
	public static void main(String[] args) throws IOException{
		Scanner scanner = new Scanner(System.in);
		int n1 = scanner.nextInt();//阶数
		int n2 = scanner.nextInt();//幂指数
        int a [][] = new int [n1][n1];//原始矩阵
        int b [][];
        int c [][] = new int [n1][n1];
        for(int i = 0;i<n1;i++)
        {
        	for(int j = 0;j<n1;j++)
        	{
        		c[i][j] = a[i][j] = scanner.nextInt();////将原始矩阵a初值赋给矩阵c
        	}
        }
        if(n2==0)//对幂为0的判断
        {
        	for (int i = 0; i < n1; i++) 
        	{
                for (int j = 0; j < n1; j++) 
                {
                    if(i == j)
                    {
                        a[i][j] = 1;
                    }
                    else
                    {
                        a[i][j] = 0;
                    }
                }
            }
        }
        while (n2 > 1) 
        {
            b = new int[n1][n1];
            for (int i = 0; i < n1; i++) 
            {
                for (int j = 0; j < n1; j++) 
                {
                    for (int k = 0; k < n1; k++) 
                    {
                        b[i][j] += a[i][k]*c[k][j];  //行列式运算
                    }
                }
            }
            a = b;
            n2--;
        }
        for (int i = 0; i < n1; i++) 
        {
            for (int j = 0; j < n1; j++) 
            {
                System.out.print(a[i][j]+" ");
            }
            System.out.println();
        }
    }         
}