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

矩阵乘法计算量估算

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

import java.util.Scanner;
import java.util.Deque;
import java.util.LinkedList;

// 注意类名必须为 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 a = in.nextInt();
            // int b = in.nextInt();
            // System.out.println(a + b);
            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(); // 计算公式 A(BC)
            Deque<Integer> queue = new LinkedList<>(); // 存放矩阵的行数和列数
            int sum = 0;
            for (int i = 0, j = 0; i < s.length(); i++) {
                if (s.charAt(i) >= 'A' &&
                        s.charAt(i) <= 'Z') { // 属于字母则把相应的行数、列数加入队列
                    queue.offer(a[j][0]);
                    queue.offer(a[j][1]);
                    j++;
                } else if (s.charAt(i) == ')') {
				  // 从 last 依次取出
                    int x0 = queue.pollLast(), y0 = queue.pollLast();
                    int x1 = queue.pollLast(), y1 = queue.pollLast();
                    sum += x0 * y0 * y1;
                    queue.offer(y1); // 存入时,要遵从FIFO 的基本规则
                    queue.offer(x0);
                }
            }
            System.out.println(sum);
        }
    }
}

双向队列有时候可以解决很多问题,尤其适合数据存储到容器后又要逆向取出来的场景。比如 LRUCache

全部评论

相关推荐

点赞 评论 收藏
分享
把实习生当正职使昨天第一天就加班,晚上连口饭都没吃上,以后日子咋过,我不想干了
码农索隆:实习不怕忙,就怕干的活重复且没难度,要干就干那种有深度有难度的任务,这样才能快速的提升
点赞 评论 收藏
分享
zYvv:双一流加大加粗再标红,然后广投。主要是获奖荣誉不够,建议开始不用追求大厂,去别的厂子刷下实习。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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