#include <iostream>
#include <stack>
#include <map>
#include <string>
using namespace std;
typedef struct Matrix
{
int a;
int b;
}Matrix;
int main()
{
int n;
cin >> n;
map<char, Matrix> m;
for (int i = 0;i < n;i++)
{
char name;
cin >> name;
Matrix matrix;cin >> matrix.a >> matrix.b;
m[name] = matrix;
}
string expr;
while (cin >> expr)
{
stack<Matrix> s;
bool error = false;
int ans = 0;
for (int i = 0;i < expr.length();i++)
{
if (isalpha(expr[i])) s.push(m[expr[i]]);
else if (expr[i] == ')')
{
Matrix m2 = s.top();s.pop();
Matrix m1 = s.top();s.pop();
if (m1.b != m2.a) { error = true;break; }
ans += m1.a*m1.b*m2.b;
Matrix m3;m3.a = m1.a;m3.b = m2.b;
s.push(m3);
}
}
if (error)cout << "error";
else cout << ans;
cout << endl;
}
return 0;
}
代码参考的紫书
分析一下,解法类似波兰式。
将矩阵存到map中(紫书是存到哈希数组了)
一个表达式是一个string,遍历string,遇到矩阵(即大写字母)就进栈,遇到右括号说明一个计算完毕,就出栈两个矩阵,先判断是否error,注意第一个出栈的是矩阵2,第二个是矩阵1,如果可以计算,就加上这次的乘法次数,并把两个矩阵相乘得到的新矩阵入栈。