算法提高 矩阵相乘

147 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

问题描述

小明最近在为线性代数而头疼,线性代数确实很抽象(也很无聊),可惜他的老师正在讲这矩阵乘法这一段内容。
当然,小明上课打瞌睡也没问题,但线性代数的习题可是很可怕的。
小明希望你来帮他完成这个任务。

现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵,
要你求出他们相乘的积(当然也是矩阵)。(输入数据保证aj=bi,不需要判断)

输入格式

输入文件共有ai+bi+2行,并且输入的所有数为整数(longlong范围内)。
第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

样例输出

19712356
6030 7141

参考代码


import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int m = in.nextInt();
        int s = in.nextInt();
        int n = in.nextInt();
        int[][] M = new int[m][s];
        int[][] N = new int[s][n];
        for ( int i = 0 ; i < m ; i++){
            for ( int j = 0 ; j < s ; j++){
                M[i][j] = in.nextInt();
            }
        }
        for ( int i = 0 ; i < s ; i++){
            for ( int j = 0 ; j < n ; j++){
                N[i][j] = in.nextInt();
            }
        }
        int[][] C = new int[m][n];
        for ( int i = 0 ; i < m ; i++){
            for (int j = 0 ; j < s ; j++){
                for ( int k = 0 ; k < n ; k++){
                    C[i][k] += M[i][j] * N[j][k];
                }
            }
        }
        for ( int i = 0 ; i < m ; i++){
            int cnt = 0;
            System.out.print(C[i][0]);
            for ( int j = 1 ; j < n ; j++){
                System.out.print(" "+C[i][j]);
            }
            if ( cnt != m-1){
                System.out.println();
                cnt++;
            }
        }
        in.close();
    }
}