UVA-442 Matrix Chain Multiplication

来源：互联网发布：unity3d 粒子系统火焰编辑：程序博客网时间：2024/06/05 01:14

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

题意：输入n个矩阵的维度和一些矩阵链乘表达式，输出乘法的次数。若乘法无法进行，输出error。假定A是m*n矩阵，B是n*p矩阵，那么AB是n*p矩阵，乘法次数为m*n*p。若A的列数不等于B的列数，则乘法无法进行。

分析：用栈即可求出结果，遇到字母入栈，遇到右括号出栈并计算，结果入栈。

#include <iostream>#include <string>#include <cstring>#include <cstdio>#include <algorithm>#include <stack>#include <vector>using namespace std;const int MAXN=30;struct rec {    int a,b;} m[MAXN];int main() {    //freopen("in.txt","r",stdin);    int n;    cin>>n;    for(int i=0; i<n; i++) {        char c;        cin>>c;        int k=c-'A';        cin>>m[k].a>>m[k].b;    }    string expr;    while(cin>>expr) {        int ans=0;        stack<rec> s;        bool error=false;        for(int i=0; i<expr.size(); i++) {            if(isalpha(expr[i])) s.push(m[expr[i]-'A']);            else if(expr[i]==')') {                rec m1=s.top();                s.pop();                rec m2=s.top();                s.pop();                if(m1.a!=m2.b) {                    error=true;                    break;                }                ans+=m2.a*m2.b*m1.b;                m2.b=m1.b;                s.push(m2);            }        }        if(!error) cout<<ans<<endl;        else  cout<<"error"<<endl;    }    return 0;}

阅读全文

0 0