Matrix Chain Multiplication UVA - 442

最新推荐文章于 2022-11-21 14:59:18 发布

努力的气球

最新推荐文章于 2022-11-21 14:59:18 发布

阅读量6.4k

点赞数 2

分类专栏：紫书

本文链接：https://blog.csdn.net/zhiyeegao/article/details/82973168

版权

紫书专栏收录该内容

9 篇文章 0 订阅

订阅专栏

Suppose you have to evaluate an expression like A*B*C*D*E where A,B,C,D and E are matrices.

Since matrix multiplication is associative, the order in which multiplications are performed is arbitrary.

However, the number of elementary multiplications needed strongly depends on the evaluation order

you choose.

For example, let A be a 50*10 matrix, B a 10*20 matrix and C a 20*5 matrix. There are two

different strategies to compute A*B*C, namely (A*B)*C and A*(B*C).

The first one takes 15000 elementary multiplications, but the second one only 3500.

Your job is to write a program that determines the number of elementary multiplications needed

for a given evaluation strategy.

Input

Input consists of two parts: a list of matrices and a list of expressions.

The first line of the input file contains one integer

(

), representing the number of

matrices in the first part. The next

lines each contain one capital letter, specifying the name of the

matrix, and two integers, specifying the number of rows and columns of the matrix.

The second part of the input file strictly adheres to the following syntax (given in EBNF):

SecondPart = Line { Line } <EOF>

Line

= Expression <CR>

Expression = Matrix | "(" Expression Expression ")"

Matrix = "A" | "B" | "C" | ... | "X" | "Y" | "Z"

Output

For each expression found in the second part of the input file, print one line containing the word ‘

error

’

if evaluation of the expression leads to an error due to non-matching matrices. Otherwise print one

line containing the number of elementary multiplications needed to evaluate the expression in the way

specified by the parentheses.

Sample Input

A 50 10

B 10 20

C 20 5

D 30 35

E 35 15

F 15 5

G 5 10

H 10 20

I 20 25

(AA)

(AB)

(AC)

(A(BC))

((AB)C)

(((((DE)F)G)H)I)

(D(E(F(G(HI)))))

((D(EF))((GH)I))

Sample Output

error

10000

error

3500

15000

40500

47500

15125

题意：对栈的简单运用，注意后进先出，还要注意别忘了写构造函数。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<deque>
#include<queue>
#include<stack>
#include<algorithm>
#include<cstring>
#include<cctype>
using namespace std;

const int maxn = 10005;
struct Matrix{
    int a,b;
    Matrix(int a=0,int b=0):a(a),b(b){}//此处的构造函数必须写（好习惯）
}m[27];

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;
}