[Java] 蓝桥杯ADV-208 算法提高矩阵相乘

问题描述
小明最近在为线性代数而头疼，线性代数确实很抽象（也很无聊），可惜他的老师正在讲这矩阵乘法这一段内容。
当然，小明上课打瞌睡也没问题，但线性代数的习题可是很可怕的。
小明希望你来帮他完成这个任务。
现在给你一个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

package adv208;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException{
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        String[] temp = bufferedReader.readLine().split(" ");
        int ai = Integer.parseInt(temp[0]);
        int aj = Integer.parseInt(temp[1]);
        long[][] a = new long[ai][aj];
        for (int i = 0; i < ai; i++) {
            String[] string = bufferedReader.readLine().split(" ");
            for (int j = 0; j < aj; j++) {
                a[i][j] = Long.parseLong(string[j]);
            }
        }

        temp = bufferedReader.readLine().split(" ");
        int bi = Integer.parseInt(temp[0]);
        int bj = Integer.parseInt(temp[1]);
        long[][] b = new long[bi][bj];
        for (int i = 0; i < bi; i++) {
            String[] strings = bufferedReader.readLine().split(" ");
            for (int j = 0; j < bj; j++) {
                b[i][j] = Long.parseLong(strings[j]);
            }
        }
        bufferedReader.close();

        for (int i = 0; i < ai; i++) {
            for (int j = 0; j < bj; j++) {
                long multipy = 0;
                for (int k = 0; k < aj; k++) {
                    multipy += a[i][k] * b[k][j];
                }
                System.out.print(multipy + " ");
            }
            System.out.println();
        }
    }

}

package adv208;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStreamReader;

public class Main {

public static void main(String[] args) throws IOException{

BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));

String[] temp = bufferedReader.readLine().split(" ");

int ai = Integer.parseInt(temp[0]);

int aj = Integer.parseInt(temp[1]);

long[][] a = new long[ai][aj];

for (int i = 0; i < ai; i++) {

String[] string = bufferedReader.readLine().split(" ");

for (int j = 0; j < aj; j++) {

a[i][j] = Long.parseLong(string[j]);

}

temp = bufferedReader.readLine().split(" ");

int bi = Integer.parseInt(temp[0]);

int bj = Integer.parseInt(temp[1]);

long[][] b = new long[bi][bj];

for (int i = 0; i < bi; i++) {

String[] strings = bufferedReader.readLine().split(" ");

for (int j = 0; j < bj; j++) {

b[i][j] = Long.parseLong(strings[j]);

}

bufferedReader.close();

for (int i = 0; i < ai; i++) {

for (int j = 0; j < bj; j++) {

long multipy = 0;

for (int k = 0; k < aj; k++) {

multipy += a[i][k] * b[k][j];

}

System.out.print(multipy + " ");

}

System.out.println();

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版