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

矩阵乘法计算量估算

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

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.Stack;


// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            int n = in.nextInt();
            int[][] arr = new int[n][2];
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < 2; j++) {
                    arr[i][j] = in.nextInt();
                }
            }
            String express = in.next();
            solution(arr, express);
        }
    }

    private static void solution(int[][] arr, String express) {

        Stack<int[]> stack = new Stack<>();
        int res = 0;
        int count = 0;
        for (int i = 0; i < express.length(); i++) {
            if (express.charAt(i) == '(') {
                continue;
            } else if (express.charAt(i) == ')') {
                int[] a1 = stack.pop();
                int[] a2 = stack.pop();
                // a2为前面的数组
                int[] newArr = {a2[0], a1[1]};
                res += a2[0] * a2[1] * a1[1];
                stack.push(newArr);
            } else {
                stack.push(arr[count++]);
            }
        }
        if (stack.size() == 1) {
            System.out.println(res);
            return;
        }

        List<int[]> list = new ArrayList<>();
        while (stack.size() > 0) {
            list.add(stack.pop());
        }
        int[] tmp = list.get(list.size() - 1);
        for (int i = list.size() - 2; i >= 0; i--) {
            int[] a = list.get(i);
            res += tmp[0] * tmp[0] * a[1];
            tmp = new int[] {tmp[0], a[1]};
        }

    }
}

全部评论

相关推荐

03-19 21:39
门头沟学院 Java
Data_Seven:6 他说的 全是我的词儿啊
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
正在热议
更多
# 春招至今,你的战绩如何? #
9396次浏览 87人参与
# 你的实习产出是真实的还是包装的? #
1707次浏览 40人参与
# 米连集团26产品管培生项目 #
5677次浏览 214人参与
# 军工所铁饭碗 vs 互联网高薪资,你会选谁 #
7424次浏览 43人参与
# 简历第一个项目做什么 #
31529次浏览 327人参与
# 重来一次,我还会选择这个专业吗 #
433325次浏览 3926人参与
# MiniMax求职进展汇总 #
23788次浏览 308人参与
# 当下环境,你会继续卷互联网,还是看其他行业机会 #
186945次浏览 1120人参与
# 牛客AI文生图 #
21408次浏览 238人参与
# 不考虑薪资和职业,你最想做什么工作呢? #
152282次浏览 887人参与
# 研究所笔面经互助 #
118864次浏览 577人参与
# 简历中的项目经历要怎么写? #
310019次浏览 4189人参与
# AI时代,哪些岗位最容易被淘汰 #
63371次浏览 801人参与
# 面试紧张时你会有什么表现? #
30482次浏览 188人参与
# 你今年的平均薪资是多少? #
213000次浏览 1039人参与
# 你怎么看待AI面试 #
179828次浏览 1232人参与
# 高学历就一定能找到好工作吗? #
64303次浏览 620人参与
# 你最满意的offer薪资是哪家公司? #
76429次浏览 374人参与
# 我的求职精神状态 #
447971次浏览 3128人参与
# 正在春招的你,也参与了去年秋招吗? #
363215次浏览 2637人参与
# 腾讯音乐求职进展汇总 #
160572次浏览 1110人参与
# 校招笔试 #
470247次浏览 2962人参与
牛客网
牛客网在线编程
牛客网题解
牛客企业服务