Java实现 蓝桥杯 算法提高 矩阵相乘

算法提高 矩阵相乘

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

问题描述

　　小明最近在为线性代数而头疼，线性代数确实很抽象（也很无聊），可惜他的老师正在讲这矩阵乘法这一段内容。

　　当然，小明上课打瞌睡也没问题，但线性代数的习题可是很可怕的。

　　小明希望你来帮他完成这个任务。

现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵，

　　要你求出他们相乘的积（当然也是矩阵）。

　　(输入数据保证aj=bi,不需要判断)

输入格式

　　输入文件共有ai+bi+2行，并且输入的所有数为整数（long long范围内）。

　　第1行：ai 和 aj

　　第2~ai+2行：矩阵a的所有元素

　　第ai+3行：bi 和 bj

　　第ai+3~ai+bi+3行：矩阵b的所有元素

输出格式

　　输出矩阵a和矩阵b的积（矩阵c)

　　（ai行bj列）

样例输入

2 2

12 23

45 56

2 2

78 89

45 56

样例输出

1971 2356

6030 7141

import java.util.Scanner;

public class 矩阵相乘 {
	public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int m = scanner.nextInt();
        int s = scanner.nextInt();
       // int n = scanner.nextInt();
        int[][] a = new int[200][200];
        int[][] b = new int[200][200];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < s; j++) {
                a[i][j] = scanner.nextInt();
            }
        }
        //因为aj=bi，其实这里完全是多余的
         s = scanner.nextInt();
        int n = scanner.nextInt();
        for (int i = 0; i < s; i++) {
            for (int j = 0; j < n; j++) {
                b[i][j] = scanner.nextInt();
            }
        }

        int[][] c = new int[200][200];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                for (int j2 = 0; j2 < s; j2++) {
                    c[i][j] += a[i][j2]*b[j2][j];
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                System.out.print(c[i][j]+" ");
            }
            System.out.println();
        }
    }

}