uva 442 Matrix Chain Multiplication

//此题其实不是考查矩阵乘法的问题，只是借这个背景来考查栈的运用，注意处理一些细节即可

 

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define MAX 30
struct matrix
{int r,c; char e;}a[MAX];
int n;
char s[10000010],stack[10000010];

int cmp(struct matrix p ,struct matrix q)
{ return p.e<q.e; }

int main()
{
    int i,j,k,top,len,sum,c;

    scanf("%d",&n); getchar();
    for(i=0; i<n; i++)
    { scanf("%c%d%d",&a[i].e,&a[i].r,&a[i].c); getchar(); }
    sort(a,a+n,cmp);

//    for(i=0; i<n; i++)
//        printf("%c %d %d\n",a[i].e,a[i].r,a[i].c);
    
while(scanf("%s",s)!=EOF)
    {
        len=strlen(s);
        if(len==1 && (s[0]>='A' && s[0]<='Z' )) 
        {printf("0\n"); continue;}

        for(sum=0,i=0,top=-1; i<len; )
        {
            if(s[i]=='(')  //左括号入栈,s移向下一位
              stack[++top]=s[i++];
            else if(s[i]>='A' && s[i]<='Z') //字母入栈
                stack[++top]=s[i++];
            else if(s[i]==')')
            {
                stack[++top]=s[i];  //先把右括号入栈，放不放其实意义不大

                c=1; 
                for(j=0; j<n; j++) if(a[j].e==stack[top-2]) break;
                for(k=0; k<n; k++) if(a[k].e==stack[top-1]) break;
                if(a[j].c!=a[k].r) break;

                c=a[j].r*a[j].c*a[k].c;  sum+=c;  //计算相乘次数

                a[n].e=a[n-1].e+1; a[n].r=a[j].r; a[n].c=a[k].c; 
                //每次相乘，就看作是产生了一个新的矩阵，把新的矩阵放入矩阵数组
                stack[top-3]=a[n].e; top-=3;
                //保留下新的矩阵在栈中
                n++; i++;
            }
        }
        if(i<len) printf("error\n");
        else      printf("%d\n",sum);
    }
    return 0;
}