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

矩阵乘法计算量估算

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

import java.util.*;
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while(in.hasNextInt()){
            int n = in.nextInt();
            int a[][] = new int[n][2];
            for(int i=0;i<n;i++){
                a[i][0] = in.nextInt();
                a[i][1] = in.nextInt();
            }
            String s = in.next();
            Stack<Integer> stack = new Stack();                 // 存放矩阵行数和列数
            int sum = 0;
            for(int i=s.length()-1,j=n-1;i>=0;i--){
                if(s.charAt(i)>='A' && s.charAt(i)<='Z'){       // 属于字母则把相应的矩阵列数和行数入栈
                    stack.push(a[j][1]);
                    stack.push(a[j][0]);
                    j--;
                }else if(s.charAt(i) == '('){                   // 括号:推出计算
                    int x0 = stack.pop(), y0 = stack.pop();     // 矩阵尺寸x0*y0
                    int x1 = stack.pop(), y1 = stack.pop();     // 矩阵尺寸x1*y1
                    sum += x0*y0*y1;      // 两个矩阵的乘法次数为x0*y0*y1或x0*x1*y1(其中y0==x1)
                    stack.push(y1);       // 把相乘后得到的矩阵列数入栈
                    stack.push(x0);       // 把相乘后得到的矩阵行数入栈
                }
            }
            System.out.println(sum);
        }
    }
}

全部评论
如果没有括号或括号里的矩阵不是刚好两个,那么数据是错的。【ABC->0,(ABC)->10000】
2 回复 分享
发布于 2022-05-19 17:12
明显不对
1 回复 分享
发布于 2023-02-17 19:39 广东
从后往前遍历的想法挺好的,细节有挺多问题
1 回复 分享
发布于 2022-08-26 12:29 北京
为什么正着不行
点赞 回复 分享
发布于 2023-04-30 17:43 江苏
nice!
点赞 回复 分享
发布于 2022-08-27 14:19 陕西

相关推荐

不愿透露姓名的神秘牛友
昨天 13:47
点赞 评论 收藏
分享
06-28 22:48
已编辑
广东金融学院 Java
小浪_Coding:学院本+这俩项目不是buff叠满了嘛
点赞 评论 收藏
分享
仁者伍敌:牛子这些人还会点一个自动回复,boss都不带回复的
点赞 评论 收藏
分享
评论
28
8
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务