题意大致是N个矩阵,如果要求计算的矩阵例如(AB),如果A的列等于B的行,进行:A.行*A.列*B.列这样的运算,如果A的列不等于B的行,输出error。
大致感觉和逆波兰表达式类似,用stack来进行存取会比较方便。如果计算(A(BC)),那么得先计算BC,再与A乘。可以以')'为标志,出现)则弹栈,取出C、B(注意栈顺序)。计算之后再将新的行列入栈,遇到下一个)时与A计算。
/*
注意进出栈的顺序,先进后出
*/
#include<iostream>
#include<stack>
using namespace std;
struct Matrix
{
int a,b;
Matrix(int a=0, int b=0):a(a),b(b){
}
}m[26];
stack<Matrix> s;
int main()
{
int n;
cin>>n;
while (n--)
{
char w;
cin >> w;
cin >> m[w-'A'].a >> m[w-'A'].b;
}
string ex;
while (cin >> ex)
{
int len = ex.length();
bool error = false;
int sum = 0;
for (int i=0; i<len; i++)
{
if (isalpha(ex[i]))
{
s.push(m[ex[i]-'A']);
}
else if (ex[i] == ')')
{
Matrix x = s.top(); //B
s.pop();
Matrix y = s.top(); //A
s.pop();
if (x.a != y.b)
{
error = true;
break;
}
else
{
sum += (y.a * y.b * x.b);
s.push(Matrix(y.a,x.b));
}
}
}
if (error)
{
cout << "error" << endl;
}
else
{
cout << sum << endl;
}
}
}