矩阵乘法

孤人 提交于 2019-12-12 02:41:42

flag

蓝桥杯第17天

题目介绍

问题描述
  输入两个矩阵,分别是ms,sn大小。输出两个矩阵相乘的结果。
输入格式
  第一行,空格隔开的三个正整数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 +01+(-1)*3=-3

思路

读取输入数据到两个二维数组中,再根据乘法公式进行计算即可
一次就通过(1.0s擦边,毕竟三重循环)

代码


import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input=new Scanner(System.in);
        int m=input.nextInt();
        int s=input.nextInt();
        int n=input.nextInt();
        long [][]x1=new long[m][s];
        long [][]x2=new long[s][n];
        for(int i=0;i<m;i++){//将第一个矩阵的值读取到二维数组x1
            for(int j=0;j<s;j++){
                x1[i][j]=input.nextLong();
            }
        }
        for(int i=0;i<s;i++){//将第二个矩阵的值读取到二维数组x2
            for(int j=0;j<n;j++){
                x2[i][j]=input.nextLong();
            }
        }

        long [][]res=new long[m][n];//保存乘积的二维数组
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                for(int k=0;k<s;k++){
                    res[i][j]+=x1[i][k]*x2[k][j];
                }
            }
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                System.out.print(res[i][j]+" ");
            }
            System.out.println();
        }
    }
}

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!