蓝桥杯 ALGO-86 矩阵乘法 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

参考代码

package 矩阵乘法;

import java.util.Scanner;

public class Main {
public static void main(String[] args) {
	Scanner sr = new Scanner( System.in);
	int m = sr.nextInt();
	int s = sr.nextInt();
	int n = sr.nextInt();
	//生成矩阵
	int[][] map1 = new int[m][s];
	int[][] map2 = new int[s][n];
	//输入矩阵
	for (int i = 0; i < map1.length; i++) {
		for (int j = 0; j < map1[0].length; j++) {
			map1[i][j] = sr.nextInt();
		}
	}
	for (int i = 0; i < map2.length; i++) {
		for (int j = 0; j < map2[0].length; j++) {
			map2[i][j] = sr.nextInt();
		}
	}
	//生成结果矩阵
	int[][] result  = new int[m][n];
	for (int i = 0; i < result.length; i++) {
		for (int j = 0; j < result[0].length; j++) {
			//result[i][j] = 矩阵1第i行 * 矩阵2第j列
			for (int k = 0; k < s; k++) {
				result[i][j] += map1[i][k] * map2[k][j];
			}
		}
	}
	//输出
	for (int i = 0; i < result.length; i++) {
		for (int j = 0; j < result[0].length; j++) {
			System.out.print(result[i][j]+" ");
		}System.out.println();
	}
}
}