UVA-442 Matrix Chain Multiplication

题目链接：https://vjudge.net/problem/UVA-442

题意：输入n个矩阵的维度和一些矩阵链乘表达式，输出乘法的次数。若乘法无法进行，输出error。假定A是m*n矩阵，B是n*p矩阵，那么AB是n*p矩阵，乘法次数为m*n*p。若A的列数不等于B的列数，则乘法无法进行。

分析：用栈即可求出结果，遇到字母入栈，遇到右括号出栈并计算，结果入栈。

#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <vector>
using namespace std;
const int MAXN=30;

struct rec {
    int a,b;
} m[MAXN];

int main() {
    //freopen("in.txt","r",stdin);
    int n;
    cin>>n;
    for(int i=0; i<n; i++) {
        char c;
        cin>>c;
        int k=c-'A';
        cin>>m[k].a>>m[k].b;
    }
    string expr;
    while(cin>>expr) {
        int ans=0;
        stack<rec> s;
        bool error=false;
        for(int i=0; i<expr.size(); i++) {
            if(isalpha(expr[i])) s.push(m[expr[i]-'A']);
            else if(expr[i]==')') {
                rec m1=s.top();
                s.pop();
                rec m2=s.top();
                s.pop();
                if(m1.a!=m2.b) {
                    error=true;
                    break;
                }
                ans+=m2.a*m2.b*m1.b;
                m2.b=m1.b;
                s.push(m2);
            }
        }
        if(!error) cout<<ans<<endl;
        else  cout<<"error"<<endl;
    }
    return 0;
}