码迷,mamicode.com
首页 > 其他好文 > 详细

【试题 基础练习 矩阵乘法】暴力矩阵乘法,再次了解一下矩阵

时间:2020-03-31 22:48:32      阅读:101      评论:0      收藏:0      [点我收藏+]

标签:输入格式   int   一个   put   ring   red   static   class   了解   

问题描述
  给定一个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.io.BufferedWriter;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.*;

public class Main {
    public  static  void main(String [] args){
        Scanner sc =new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int arr[][] = new int[n+10][n+10];
        int brr[][] = new int[n+10][n+10];
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                brr[i][j]=arr[i][j]=sc.nextInt();
        if(m==0) {
            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 if(m==1){
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++) System.out.print(arr[i][j]+" ");
                System.out.println();
            }

        }else {
             for(int z=1;z<m;z++){
                 int temp[][]=new int[n+10][n+10];
                 for(int i=0;i<n;i++)
                   for(int j=0;j<n;j++){
                     int add = 0;
                        for(int y=0;y<n;y++)
                           add+=arr[i][y]*brr[y][j];
                        temp[i][j]=add;
                  }
                  brr=temp;
             }
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++) System.out.print(brr[i][j]+" ");
                System.out.println();
            }
        }

   }
}

  

 

【试题 基础练习 矩阵乘法】暴力矩阵乘法,再次了解一下矩阵

标签:输入格式   int   一个   put   ring   red   static   class   了解   

原文地址:https://www.cnblogs.com/wszhu/p/12609134.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!