UVa - 442 Matrix Chain Multiplication(栈模拟)

题目链接:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=19085

#include <iostream>
#include <stack>

using namespace std;
/*************************************************************************************************************
        题意:输入n个矩阵的维度和一些矩阵链乘表达式,输出乘法的次数。假定A和m*n的,B是n*p的,
              那么AB是m*p的,乘法次数为m*n*p
        思路:
        1,利用栈模拟,碰到字母入栈,碰到右括号 ')' 出栈两个元素,计算并入栈
        2,利用结构体实现矩阵行列两个变量的入栈
        3,出栈的同时判断是否满足计算的条件(即 m1.y == m2.x)否则直接退出输出 'error'
        4,学到一个技巧。进栈方式
            a,利用结构体变量入栈,如 39 行
            b,利用结构体的构造函数实现多变量的进栈,如 49 行
*************************************************************************************************************/

struct Node{
    int x,y;
    Node(int x=0,int y=0) : x(x),y(y) {}
}node[30];

int main()
{
    int n;
    cin>>n;

    for(int i = 1;i <= n;i ++){
        string c;
        cin>>c;
        int k=c[0]-'A';
        cin>>node[k].x>>node[k].y;
    }

    string s;
    while(cin>>s){
        int flag=1,ans=0;
        stack<Node> car;
        for(int i = 0;i < s.size();i ++){
            if(isalpha(s[i]))
                car.push(node[s[i]-'A']);
            else if(s[i] == ')'){
                Node m2=car.top();  car.pop();
                Node m1=car.top();  car.pop();
                if(m2.x != m1.y){
                    flag=0;
                    break;
                }
                ans+=m1.x*m1.y*m2.y;
                car.push(Node(m1.x,m2.y));
            }
        }
        if(flag)
            cout<<ans<<endl;
        else
            cout<<"error"<<endl;
    }
    return 0;
}



 

posted on 2016-05-13 02:15  Jstyle  阅读(104)  评论(0编辑  收藏  举报

导航