题意:输入n个矩阵的维度和一些矩阵链乘表达式;输出乘法的次数。如果两个矩阵无法相乘,输出error。
题解:列:矩阵A维度 m*n,B 维度 n*p,则 AB 维度 m*p, 乘法的次数 m*n*p;
出现error的情况是 A的列不等于B的行;我们可以用栈 存储矩阵链乘表达式的字母,遇到右括号出栈计算,并将计算的结果入栈;
#include<iostream>
#include<cstring>
#include<stack>
using namespace std;
struct Matrix{
int x,y;
Matrix(int x = 0, int y = 0):x(x),y(y) {}//构造函数
}m[26];
stack<Matrix> s;
int main()
{
int n;
cin >> n;
for (int i = 0; i < n; i++)
{
string name;
cin >> name;
int k = name[0] - 'A';
cin >> m[k].x >> m[k].y;
}
string expr;
while(cin >> expr)
{
int len = expr.length();
bool error = false;
int ans = 0;
for (int i = 0; i < len; i++)
{
if(isalpha(expr[i])) s.push(m[expr[i]-'A']);
else if (expr[i] == ')')
{
Matrix m2 = s.top(); s.pop();
Matrix m1 = s.top(); s.pop();
if(m1.y != m2.x) { error = true; break;}
ans += m1.x * m1.y * m2.y;
s.push(Matrix(m1.x, m2.y));//将结果入栈
}
}
if(error) printf("error\n");
else printf("%d\n", ans);
}
return 0;
}