Matrix Chain Multiplication (堆栈)

题目链接:https://vjudge.net/problem/UVA-442

题目大意:输入n个矩阵的维度和一些矩阵链乘表达式,输出乘法的次数。如果乘法无法进行,输出error。 假定A是m*n的矩阵,B是n*p的矩阵,乘法次数为m*n*p。如果A的列数不等于B的行数,则乘法

无法进行。

  例如A是50*10的,B是10*20的,C是20*5的,则(A(BC))的乘法次数为10*20*5(BC的乘法次数)+50*10*5((A(BC)的乘法次数)=3500

分析:本题的关键是解析表达式,本题的表达式比较简单,可以用一个栈来完成,遇到字母时入栈,遇到右括号时出栈并计算,然后结果入栈。 因为保证输入合法,括号无需入栈

#include<iostream>
#include<stack>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=26+5;
struct Matrix
{
    int a,b;
    //Matrix (int a=0,int b=0):a(a),b(b){}
    Matrix (int c=0,int d=0)
    {
        a=c;
        b=d;
    }
}m[maxn];
stack<Matrix> s;
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        string name;
        cin>>name;
        int k=name[0]-'A';//存下标
        cin>>m[k].a>>m[k].b;
    }
    string expr;
    while(cin>>expr)
    {
        int len=expr.length();
        bool error=false;
        int ans=0;
        for(int i=0;i<len;i++)
        {
            if(isalpha(expr[i])) s.push(m[expr[i]-'A']);//是否是字母  也就是矩阵  是的话入栈
            else if(expr[i]==')')
            {
                Matrix m2=s.top(); s.pop();//取两个字符
                Matrix m1=s.top(); s.pop();
                if(m1.b!=m2.a)
                {
                    error=true;
                    break;
                }
                ans+=m1.a*m1.b*m2.b;
                s.push(Matrix(m1.a,m2.b));
            }
        }
        if(error) cout<<"error"<<endl;
        else cout<<ans<<endl;
    }
    return 0;
}

 

posted @ 2019-01-31 10:38  执||念  阅读(471)  评论(0编辑  收藏  举报