题解 | #矩阵乘法计算量估算#

矩阵乘法计算量估算

https://www.nowcoder.com/practice/15e41630514445719a942e004edc0a5b

import java.util.*;

/**
 * @author hll[yellowdradra@foxmail.com]
 * @date 2022-10-13 21:36
 **/
public class Main {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        Matrix[] matrix = new Matrix[in.nextInt()];
        fill(matrix, in);
        String s = in.nextLine();
        LinkedList<Matrix> stack = new LinkedList<Matrix>();
        int multiplications = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '(') {
                continue;
            } else if (s.charAt(i) == ')') {
                Matrix a = stack.pop();
                Matrix b = stack.pop();
                // 矩阵乘法是不满足交换律的 注意出栈顺序
                multiplications += b.multiply(a);
                stack.push(b);
            } else {
                stack.push(matrix[s.charAt(i) - 'A']);
            }
        }
        System.out.println(multiplications);
    }

    static void fill(Matrix[] arr, Scanner in) {
        for (int i = 0; i < arr.length; i++) {
            arr[i] = new Matrix(in.nextInt(), in.nextInt());
        }
        // 吞一个回车 要不还要转换 有点麻烦
        in.nextLine();
    }

    static class Matrix {
        int x, y;

        Matrix(int x, int y) {
            this.x = x;
            this.y = y;
        }

        int multiply(Matrix b) {
            int t = this.x * this.y * b.y;
            this.y = b.y;
            return t;
        }
    }
}

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-24 20:55
阿里国际 Java工程师 2.7k*16.0
程序员猪皮:没有超过3k的,不太好选。春招再看看
点赞 评论 收藏
分享
点赞 评论 收藏
分享
10-27 17:26
东北大学 Java
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务