算法训练 矩阵乘法

    xiaoxiao2021-12-15  8

      算法训练 矩阵乘法   时间限制: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)的值。 样例输入 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.*; public class Main { public static void main(String args[]){ Scanner in=new Scanner(System.in); int m=in.nextInt(); int s=in.nextInt(); int n=in.nextInt(); int[][] a=new int[m][s]; for(int i=0;i<m;i++){//第一个矩阵 for(int j=0;j<s;j++){ a[i][j]=in.nextInt(); } } int[][] b=new int[s][n]; for(int i=0;i<s;i++){//第二个矩阵 for(int j=0;j<n;j++){ b[i][j]=in.nextInt(); } } int[][] c=new int[m][n]; //矩阵相乘 for(int i=0;i<m;i++){//行 for(int j=0;j<n;j++){//列 c[i][j]=0; 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++){ if(j!=n-1){ System.out.print(c[i][j]+" "); }else{ System.out.print(c[i][j]); } } System.out.println(); } } }

    转载请注明原文地址: https://ju.6miu.com/read-1000270.html

    最新回复(0)