算法训练 矩阵乘法

92 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

问题描述

  输入两个矩阵,分别是ms,sn大小。输出两个矩阵相乘的结果。

输入格式

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

输出格式

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

样例输入

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

样例输出

-32
-8 2

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

参考代码

import java.util.Scanner;

public class Main {

    publicstatic void main(String[] args) {
       Scanner in = newScanner(System.in);
       int m =in.nextInt();
       int s =in.nextInt();
       int n =in.nextInt();
       int[][] M = newint[m][s];
       int[][] N = newint[s][n];
       for ( int i = 0 ; i < m ;i++){
          for ( int j = 0 ; j < s ; j++){
              M[i][j] =in.nextInt();
          }
       }
       for ( int i = 0 ; i < s ;i++){
          for ( int j = 0 ; j < n ; j++){
              N[i][j] =in.nextInt();
          }
       }
       int[][] C = newint[m][n];
       for ( int i = 0 ; i < m ;i++){
          for (int j = 0 ; j < s ; j++){
              for ( intk = 0 ; k < n ; k++){
                 C[i][k] += M[i][j] *N[j][k];
             }
          }
       }
       for ( int i = 0 ; i < m ;i++){
          int cnt = 0;
          System.out.print(C[i][0]);
          for ( int j = 1 ; j < n ; j++){
             System.out.print(" "+C[i][j]);
          }
          if ( cnt != m-1){
             System.out.println();
             cnt++;
          }
   }
}