UVA442 Matrix Chain Multiplication 矩阵运算量计算（栈的简单应用）

栈的练习，如此水题竟然做了两个小时。。。

题意：给出矩阵大小和矩阵的运算顺序，判断能否相乘并求运算量。

我的算法很简单：比如(((((DE)F)G)H)I)，遇到 (就cnt累计加一，字母入栈，遇到)减一，并出栈两个矩阵计算运算量，将计算后的矩阵压入栈。当cnt等于0时就输出运算量。

难点是当不能运算后的处理。

卡那么就其实主要是细节问题，最大的坑是里面退栈时倒着退出，没注意到结果每次计算都判断为不能计算。。。

AC代码：

#include <iostream>
#include <cstdio>
#include <stack>
using namespace std;
int const maxn = 27;

struct Mat{
int x, y;
};
Mat mat[maxn];

Mat multip(Mat a, Mat b) {
Mat tmp;
tmp.x = a.x;
tmp.y = b.y;
return tmp;
}

int main() {
int n, cnt = 0, kh = 0;
bool flag = true;
char tmp;
Mat a, b;
stack <Mat> v;
freopen("in", "r", stdin);
cin >> n;
while (n--) {
cin >> tmp;
tmp -= 'A';
cin >> mat[tmp].x >> mat[tmp].y;
}//while
while (cin >> tmp) {
if (kh == 0) {
flag = true;
}
if (flag == false) {
if (tmp == '(')
kh++;
else if (tmp == ')')
kh--;
continue;
}
if (tmp >= 'A' && tmp <= 'Z') {
v.push(mat[tmp - 'A']);
}//alpha
else{
if (tmp == '(') {
kh++;
}//(
else if (tmp == ')'){
kh--;
a = v.top();
v.pop();
b = v.top();
v.pop();
if (b.y != a.x) {
cout << "error" << endl;
cnt = 0;
flag = false;
continue;
}
cnt += b.x * a.x * a.y;
v.push(multip(b, a));
}//)
}//not alpha
if (kh == 0) {
cout << cnt << endl;
cnt = 0;
}//print
}//while
return 0;
}

提交时又忘记去掉文件重定向了，wa了一下。。。