蓝桥杯算法训练矩阵乘法

最新推荐文章于 2021-01-31 22:33:19 发布

dougan_

最新推荐文章于 2021-01-31 22:33:19 发布

阅读量199

点赞数

分类专栏：蓝桥杯

本文链接：https://blog.csdn.net/dougan_/article/details/78825726

版权

蓝桥杯专栏收录该内容

31 篇文章 0 订阅

订阅专栏

算法训练矩阵乘法

时间限制：1.0s 内存限制：512.0MB

问题描述

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

输入格式

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

输出格式

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

样例输入

样例输出

   -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 sc=new Scanner(System.in);
		int m=sc.nextInt();
		int s=sc.nextInt();
		int n=sc.nextInt();
		int a[][]=new int[m][s];
		int b[][]=new int[s][n];
		for(int i=0;i<m;i++) {
			for(int j=0;j<s;j++) {
				a[i][j]=sc.nextInt();
			}
		}
		for(int i=0;i<s;i++) {
			for(int j=0;j<n;j++) {
				b[i][j]=sc.nextInt();
			}
		}
		int c[][]=new int[m][n];
		for(int i=0;i<m;i++) {
			for(int j=0;j<n;j++) {
				for(int k=0;k<s;k++)
				c[i][j]+=a[i][k]*b[k][j];
			}
		}
		for(int i=0;i<m;i++) {
			for(int j=0;j<n;j++) {
				System.out.print(c[i][j]+" ");
			}
			System.out.println();
		}
}
}