蓝桥杯 矩阵乘法

资源限制

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

 

问题描述

　　小明最近刚刚学习了矩阵乘法，但是他计算的速度太慢，于是他希望你能帮他写一个矩阵乘法的运算器。

 

输入格式

　　输入的第一行包含三个正整数N,M,K，表示一个N*M的矩阵乘以一个M*K的矩阵。接下来N行，每行M个整数，表示第一个矩阵。再接下来的M行，每行K个整数，表示第二个矩阵。

 

输出格式

　　输出有N行，每行K个整数，表示矩阵乘法的结果。

 

一道很简单的题，直接看代码。

 

 

import java.util.Scanner;

public class Main {
    
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int m=scanner.nextInt();
        int n=scanner.nextInt();
        int k=scanner.nextInt();
        int[][]a=new int[m][n];
        int[][]b=new int[n][k];
        int[][]ans=new int[m][k];
        for(int i=0;i<m;i++) {
            for(int j=0;j<n;j++) {
                a[i][j]=scanner.nextInt();
            }
        }
        
        for(int i=0;i<n;i++) {
            for(int j=0;j<k;j++) {
                b[i][j]=scanner.nextInt();
            }
        }
        
        for(int i=0;i<m;i++) {
            for(int j=0;j<k;j++) {
                for(int t=0;t<n;t++) {
                    ans[i][j]+=a[i][t]*b[t][j];
                }
            }
        }
        
        for(int i=0;i<m;i++) {
            for(int j=0;j<k;j++) {
                System.out.print(ans[i][j]+" ");
            }
            System.out.println("");
        }
    }
}