蓝桥杯试题矩阵乘法java实现

最新推荐文章于 2025-02-20 22:29:29 发布

原创最新推荐文章于 2025-02-20 22:29:29 发布 · 487 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#简单易懂的新手算法 #java #class

算法试题（java）专栏收录该内容

177 篇文章

订阅专栏

该博客介绍了如何使用Java来实现矩阵乘法，特别是在解决蓝桥杯竞赛试题时的应用。文章首先创建了二维数组并读取输入数据，然后通过循环计算矩阵的n次幂。当幂为0时，输出对角线为1的单位矩阵；为1时，直接输出原矩阵；否则，使用临时数组进行多次矩阵相乘。最后，博客展示了完整的Java代码实现。

import java.util.Scanner;

public class Main {

   public static void main(String[] args) {
       Scanner sc=new Scanner(System.in);
       int a=sc.nextInt();
       int n=sc.nextInt();
       int[][] m=new int[a][a];
       int[][] f=new int[a][a];//为了保留结果多次幂时会用到
       //存入到二维数组中去
       for(int i=0;i<a;i++) {
           for(int k=0;k<a;k++) {
               f[i][k]=m[i][k]=sc.nextInt();
           }
       }
       //判断幂的次数当等于零时对角线上的数字为一
       if(n==0) {
           for(int i=0;i<a;i++) {
               for(int k=0;k<a;k++) {
                   if(i==k) {
                       System.out.print(1+" ");
                   }else {
                       System.out.print(0+" ");
                   }
               }
               System.out.println();
           }
           //1次幂时原样输出
       }else if(n==1) {
           for(int i=0;i<a;i++) {
               for(int k=0;k<a;k++) {
                   System.out.print(m[i][k]+" ");
                   }
               System.out.println();
               }
       }else {
           //几次幂几次循环
       for(int i=1;i<n;i++) {
           //保留运算结果不会改变开始建立的两个数组
           int[][] tmp = new int[a][a];
           for(int   k=0;k<a;k++) {
               for(int l=0;l<a;l++) {
                   int e=0;//用于累加储存结果
                   for(int t=0;t<a;t++) {
                       e+=m[k][t]*f[t][l];
                   }
                   tmp[k][l]=e;//赋值到tmp数组中
               }
           }
           f=tmp;//循环完毕更改保留结果以便于下次幂运算时进行累乘
       }
   //打印结果
       for(int i=0;i<a;i++) {
           for(int k=0;k<a;k++) {
               System.out.print(f[i][k]+" ");
           }
           System.out.println();
       }
       }
   }

}