1094 of ZOJ - AC的第二题

近来想起了ZOJ，登上去发现很久以前就注册的好，AC了1001，练手的那题。找了道AC率最高的题（就是1094）开始做。实在是太久太久没认真写程序了，脑子不灵光，做了大半天，这会儿才AC，杯具啊。

是矩阵相乘的一道题，唯一学的一点离散是在中学了，早忘光了。先花了若干时间弄懂矩阵相乘。

表达式的计算，上来就用递归做（真不知道当时怎么想的）。写了半天，发现越写越复杂，突然就想起了括号匹配，网上一搜，用栈做啊。（亏还参加过NOIP呢，这么简单的竟然忘了）三下两下把代码改好，一次AC。。成就感还真不小。。

不多说，贴出代码吧。。

题目：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1094

#include <iostream> #include <string> using namespace std; static int matrix[26][2]; void init() { short n=0; char ch; cin>>n; for (int i=0;i<n;i++) { cin>>ch; cin>>matrix[ch-'A'][0]>>matrix[ch-'A'][1]; } return ; } long func(string str) { long result=0; int pos=0; int stack[26][2]={0}; for (int i=0;i<str.length();i++) { if ((str[i]>='A')&&(str[i]<='Z')) { stack[pos][0]=matrix[str[i]-'A'][0]; stack[pos][1]=matrix[str[i]-'A'][1]; pos++; } if (str[i]==')') { if (stack[pos-1][0]==stack[pos-2][1]) { result+=stack[pos-1][0]*stack[pos-1][1]*stack[pos-2][0]; stack[pos-2][1]=stack[pos-1][1]; pos--; } else { return -1; } } } if (pos==2) { if (stack[pos-1][0]==stack[pos-2][1]) { result+=stack[pos-1][0]*stack[pos-1][1]*stack[pos-2][0]; } else { return -1; } } return result; } int main() { int t[2]; string str; long result; init(); while (cin>>str) { result=func(str); if (result==-1) cout<<"error"<<endl; else cout<<result<<endl; } return 0; }