矩阵乘法

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();
        }
    }
}

tip

当数据量变大时可能超时，这是因为Scanner在数据量大的时候效率低
这是java语言运行超时的一些解决办法：
https://blog.csdn.net/richenyunqi/article/details/84350768

import java.io.*;
public class Main2 {
    public static void main(String[] args) throws IOException {
        BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
        String[] s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
        int M = Integer.parseInt(s[0]), S = Integer.parseInt(s[1]), N = Integer.parseInt(s[2]);//将字符串转换为整数
        int[][] A1 = new int[M][S];
        int[][] A2 = new int[S][N];
        int[][] A3 = new int[M][N];
        for (int i = 0; i < M; ++i) {
            s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
            for (int j = 0; j < S; ++j)
                A1[i][j] = Integer.parseInt(s[j]);//将字符串转换为整数
        }
        for (int i = 0; i < S; ++i){
            s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
            for (int j = 0; j < N; ++j)
                A2[i][j] = Integer.parseInt(s[j]);//将字符串转换为整数
        }
        for (int i = 0; i < M; ++i)
            for (int j = 0; j < N; ++j) {
                A3[i][j] = 0;
                for (int k = 0; k < S; ++k)
                    A3[i][j] += A1[i][k] * A2[k][j];
            }
        for (int i = 0; i < M; ++i) {
            for (int j = 0; j < N; ++j)
                System.out.print(+A3[i][j] + " ");
            System.out.println();
        }
    }
}