HJ70 矩阵乘法计算量估算

HJ70 矩阵乘法计算量估算

描述

矩阵乘法的运算量与矩阵乘法的顺序强相关。
例如：
A 是一个 50×10 的矩阵，B 是 10×20 的矩阵，C 是 20×5 的矩阵

计算 ABC 有两种顺序：((AB)C)或者(A(BC))，前者需要计算 15000 次乘法，后者只需要 3500 次。

编写程序计算不同的计算顺序需要进行的乘法次数。

数据范围：矩阵个数：1≤n≤15 1≤n≤15 ，行列数：1≤rowi,coli≤100 1≤rowi​,coli​≤100 ，保证给出的字符串表示的计算顺序唯一。
进阶：时间复杂度：O(n) O(n) ，空间复杂度：O(n) O(n)

输入描述：

输入多行，先输入要计算乘法的矩阵个数 n，每个矩阵的行数，列数，总共 2n 的数，最后输入要计算的法则
计算的法则为一个字符串，仅由左右括号和大写字母（‘A’~‘Z’）组成，保证括号是匹配的且输入合法！

输出描述：

输出需要进行的乘法次数

题解

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        while (in.hasNext()) {
// 接收数据
            // 输入多行
            // 先输入要计算乘法的矩阵个数n，
            int n = Integer.parseInt(in.nextLine());
            // 每个矩阵的行数，列数，总共2n的数，
            int num[][] = new int[n][2];
            for (int i = 0; i < n; i++) {
                String [] temp = in.nextLine().split(" ");
                num[i][0] = Integer.parseInt(temp[0]);
                num[i][1] = Integer.parseInt(temp[1]);
            }
            // 最后输入要计算的法则
            // 计算的法则为一个字符串，仅由左右括号和大写字母（'A'~'Z'）组成，保证括号是匹配的且输入合法！
            String str = in.nextLine();
// 计算量估计
            // 栈
            Stack <int[]> stack = new Stack<>();
            int sum = 0;
            for (int i = 0; i < str.length(); i++) {
                if (str.charAt(i) == ')') {
                    // 上一个矩阵的行数和列数
                    // pop 移除堆栈顶部的对象，并作为此函数的值返回该对象。
                    // 每次遇到右括号将栈顶两个矩阵相乘 并将新矩阵重新压到栈顶
                    int [] last = stack.pop();
                    int [] near = stack.pop();
                    // last 10 20
                    // near 50 10
                    // 累计本次计算的次数 10*20*50 i *j *k
                    sum = sum + last[0] *  last[1] * near[0];
                    // 新矩阵  near[0] laxt[1] 50 20
                    stack.push(new int[] {near[0], last[1]});
                } else if (str.charAt(i) == '(') {
                } else {
                    // 压入栈顶 矩阵
                    stack.push(num[str.charAt(i) - 'A']);
                }
            }
            System.out.println(sum);
        }
    }
}