题解 | #矩阵乘法#

矩阵乘法

https://www.nowcoder.com/practice/ebe941260f8c4210aa8c17e99cbc663b

暴力破解法:
import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        while (in.hasNextInt()) { // 注意 while 处理多个 case
            int x = in.nextInt();
            int y = in.nextInt();
            int z = in.nextInt();
            int[][] A = new int[x][y];
            int[][] B = new int[y][z];
            for (int i = 0; i < x; i++) {
                for (int j = 0; j < y; j++) {
                    A[i][j] = in.nextInt();
                }
            }
            for (int i = 0; i < y; i++) {
                for (int j = 0; j < z; j++) {
                    B[i][j] = in.nextInt();
                }
            }
            int[][] C = new int[x][z];

            for (int i = 0; i < x; i++) {
                for (int j = 0; j < z; j++) {
                    for (int k = 0; k < y; k++) {
                        C[i][j] += A[i][k] * B[k][j];
                    }
                    if (j == z - 1) {
                        System.out.println(C[i][j]);
                    } else {
                        System.out.print(C[i][j] + " ");
                    }
                }
            }

        }
    }
}
使用BufferedReader来输入可以减少运行时间:
import java.util.Scanner;
import java.io.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) throws IOException {
        //Scanner in = new Scanner(System.in);
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String str = null;
        // 注意 hasNext 和 hasNextLine 的区别
        //while (in.hasNextInt()) { // 注意 while 处理多个 case
        while ((str = br.readLine()) != null) {
            // int x = in.nextInt();
            // int y = in.nextInt();
            // int z = in.nextInt();
            int x = Integer.parseInt(str);
            int y = Integer.parseInt(br.readLine());
            int z = Integer.parseInt(br.readLine());
            int[][] A = new int[x][y];
            int[][] B = new int[y][z];
            // for (int i = 0; i < x; i++) {
            //     for (int j = 0; j < y; j++) {
            //         A[i][j] = in.nextInt();
            //     }
            // }
            for (int i = 0; i < x; i++) {
                String[] params = br.readLine().split(" ");
                for (int j = 0; j < y; j++) {
                    A[i][j] = Integer.parseInt(params[j]);
                }
            }
            // for (int i = 0; i < y; i++) {
            //     for (int j = 0; j < z; j++) {
            //         B[i][j] = in.nextInt();
            //     }
            // }
            for (int i = 0; i < y; i++) {
                String[] params = br.readLine().split(" ");
                for (int j = 0; j < z; j++) {
                    B[i][j] = Integer.parseInt(params[j]);
                }
            }

            int[][] C = new int[x][z];
            for (int i = 0; i < x; i++) {
                for (int j = 0; j < z; j++) {
                    for (int k = 0; k < y; k++) {
                        C[i][j] += A[i][k] * B[k][j];
                    }
                    if (j == z - 1) {
                        System.out.println(C[i][j]);
                    } else {
                        System.out.print(C[i][j] + " ");
                    }
                }
            }
        }
    }
}



复杂度分析

时间复杂度O(n*m*k),n为A矩阵行数,m为B矩阵列数,k为B矩阵(A矩阵列数)行数

空间复杂度O(1), 直接判断,无额外空间。




全部评论

相关推荐

独玖:同二本,建议咱俩一起重开
点赞 评论 收藏
分享
下北泽:都是校友,还是同届,我就说直白点,不委婉了,我相信你应该也不是个玻璃心,首先你觉得一个双非的绩点写简历上有用吗?班长职务有用吗?ccf有用吗?企业会关心你高数满分与否吗?第二,第一个项目实在太烂,一眼就能看出是外卖,还是毫无包装的外卖,使用JWT来鉴权,把热点数据放进Redis这两个点居然还能写进简历里,说难听点这两个东西都是学个几十分钟,调用个API就能完成的事情,在双非一本的条件下,这种项目你觉得能拿出手吗,第二个项目你写的东西和你的求职方向有任何的匹配吗?第三,计设那一块毫无价值,如果想突出自己会前端,直接写入专业技能不行吗,最后,专业技能里像深入理解JVM底层原理这种你觉得这句话你自己真的能匹配吗?都是校友加上同届,我措辞直接,但希望能点出你的问题,想进大厂还得继续沉淀项目和学习
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务