UVA 442 栈解析表达式

#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，如果可以计算，就加上这次的乘法次数，并把两个矩阵相乘得到的新矩阵入栈。