HJ70 矩阵乘法计算量估算

描述

矩阵乘法的运算量与矩阵乘法的顺序强相关。
例如：

A是一个50×10的矩阵，B是10×20的矩阵，C是20×5的矩阵

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

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

数据范围：矩阵个数：1≤n≤15 ，行列数：1≤rowi​,coli​≤100 ，保证给出的字符串表示的计算顺序唯一。

进阶：时间复杂度：O(n) ，空间复杂度：O(n) 

输入描述：

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

输出描述：

输出需要进行的乘法次数

示例1

输入：

3
50 10
10 20
20 5
(A(BC))

输出：

3500

---

#include<cstdio>
#include<string>
#include<vector>
#include<stack>
using namespace std;

struct Rect{
    int row;//行
    int col;//列
};
stack<Rect> rect_stack;
int main()
{
    // 1.读取数据
        // 输入矩阵个数
    int n;
    scanf("%d",&n); 
        // 输入各个矩阵
    vector<Rect> rect;
    for(int i=0;i<n;i++)
    {
        int r,c;
        Rect temp; 
        scanf("%d %d",&r,&c);
        temp.row=r;
        temp.col=c;
        rect.push_back(temp);        
    }
        // 输入计算法则
    char buf[200];
    scanf("%s",buf);
    string str = buf;

    // 2.计算
    int count=0;
    int index=0;
    for(int i=0 ; i<str.size() ; i++)
    {
        if(str[i]>='A' && str[i]<='Z')
        {
            rect_stack.push(rect[index]);
            index++;
        }
        else if (str[i]==')') {
            // 取出两个矩阵
            Rect right=rect_stack.top();
            rect_stack.pop();
            Rect left=rect_stack.top();
            rect_stack.pop();
            // 统计乘法次数
            count+=left.row * left.col * right.col;
            // 相乘后的矩阵压入栈中
            Rect result;
            result.row=left.row;
            result.col=right.col;
            rect_stack.push(result);
        }
    }

    // 3.输出
    printf("%d",count);


}