您的位置:首页 > 其它

算法训练 矩阵乘法

2017-02-02 17:00 399 查看
问题描述

  输入两个矩阵,分别是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

import java.util.Scanner;

public class Main {

public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner in = new Scanner(System.in);
int m = in.nextInt();
int s = in.nextInt();
int n = in.nextInt();
int[][] M = new int[m][s];
int[][] N = new int[s]
;
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 = new int[m]
;
for ( int i = 0 ; i < m ; i++){
for (int j = 0 ; j < s ; j++){
for ( int k = 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++;
}
}
in.close();
}
}


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: