蓝桥杯-算法训练-矩阵乘法-Java

蓝桥杯-算法训练-矩阵乘法-Java

• 问题描述

  　　输入两个矩阵，分别是m*s，s*n大小。输出两个矩阵相乘的结果。

• 输入格式

  　　第一行，空格隔开的三个正整数m,s,n（均不超过200）。
  　　接下来m行，每行s个空格隔开的整数，表示矩阵A（i，j）。
  　　接下来s行，每行n个空格隔开的整数，表示矩阵B（i，j）。
  　　

• 输出格式

  m行，每行n个空格隔开的整数，输出相乘後的矩阵C（i，j）的值。

• 样例输入

  2 3 2
  1 0 -1
  1 1 -3
  0 3
  1 2
  3 1

• 样例输出

  -3 2
  -8 2

提示
矩阵C应该是m行n列，其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1)(0,1,3) = 1 0 +0*1+(-1)*3=-3

思考

在第一个矩阵的列数和第二个矩阵的行数相等时才能实现矩阵的相乘，这里我定义二维数组来表示矩阵，通过for循环来进行矩阵的相乘。

代码

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int m = 0 , s = 0 , n = 0 ;
        if(scanner.hasNext()){  //输入m,s,n
            m = scanner.nextInt() ;
            s = scanner.nextInt() ;
            n = scanner.nextInt() ;
        }
        //定义两个矩阵,行列数分别为m,s与s,n
        int arr1[][] = new int[m][s] ;
        int arr2[][] = new int[s][n] ;
        for(int i = 0 ; i < m ; i ++){  //输入第一个矩阵
            for(int j = 0 ; j < s ; j++){
                if(scanner.hasNext()){
                    arr1[i][j] = scanner.nextInt() ;
                }
            }
        }
        for(int i = 0 ; i < s ; i++){   //输入第二个矩阵
            for(int j = 0 ; j < n ; j++){
                if(scanner.hasNext()){
                    arr2[i][j] = scanner.nextInt() ;
                }
            }
        }
        int arr[][] = new int[m][n] ;   //定义一个相乘后的矩阵，矩阵的行数与列数为m，n
        for(int i = 0 ; i < m ; i++){   //进行矩阵相乘
            for(int j = 0 ; j < n ; j++){
                for(int x = 0 ; x < s ; x ++){  //arr[i][j] = arr1[i][1] * arr2[1][j] + ... + arr1[i][s-1] * arr2[s-1][j] 
                    arr[i][j] += arr1[i][x] * arr2[x][j] ;
                }
                System.out.print(arr[i][j] + " ");
            }
            System.out.println();
        }
    }
}