码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2246 Matrix Chain Multiplication

时间:2015-08-08 22:34:24      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:

用栈来处理一下就好了。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<map>
using namespace std;

struct Mar
{
    int r,c;
} mar[3000000];
int n;
char s[100000];
int r,c,top,flag;
Mar Stack[1000000];

int main()
{
    scanf("%d",&n);
    for(int i=0; i<n; i++)
    {
        scanf("%s%d%d",s,&r,&c);
        mar[s[0]-A].r=r;
        mar[s[0]-A].c=c;
    }
    while(~scanf("%s",s))
    {
        int len=strlen(s);
        flag=0;
        top=-1;
        for(int i=0; i<len; i++)
        {
            if(top==-1)
            {
                top++;
                if(s[i]>=A&&s[i]<=Z)
                    Stack[top]=mar[s[i]-A];
                else
                {
                    Mar node;
                    node.c=-1;
                    node.r=-1;
                    Stack[top]=node;
                }
                continue;
            }
            if(s[i]==()
            {
                Mar node;
                node.c=-1;
                node.r=-1;
                top++;
                Stack[top]=node;
            }
            else if(s[i]==))
            {
                top--;
                Stack[top]=Stack[top+1];
                if(top-1!=-1&&Stack[top-1].c!=-1&&Stack[top-1].r!=-1)
                {
                    if(Stack[top-1].c!=Stack[top].r)
                    {
                        flag=-1;
                        break;
                    }
                    else
                    {
                        flag=flag+Stack[top-1].c*Stack[top-1].r*Stack[top].c;
                        Stack[top-1].c=Stack[top].c;
                        top--;
                    }
                }
            }
            else
            {
                Mar node;
                node.c=mar[s[i]-A].c;
                node.r=mar[s[i]-A].r;
                top++;
                Stack[top]=node;
                if(Stack[top-1].r!=-1&&Stack[top-1].c!=-1)
                {
                    if(Stack[top-1].c!=Stack[top].r)
                    {
                        flag=-1;
                        break;
                    }
                    else
                    {
                        flag=flag+Stack[top-1].c*Stack[top-1].r*Stack[top].c;
                        Stack[top-1].c=Stack[top].c;
                        top--;
                    }
                }
            }
        }
        if(flag==-1) printf("error\n");
        else printf("%d\n",flag);
    }
    return 0;
}

 

POJ 2246 Matrix Chain Multiplication

标签:

原文地址:http://www.cnblogs.com/zufezzt/p/4713952.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!