UVa - 442 Matrix Chain Multiplication(栈模拟)

最新推荐文章于 2022-01-02 20:07:43 发布

转载最新推荐文章于 2022-01-02 20:07:43 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Jstyle-continue/p/6351973.html

本文介绍了一种使用栈来模拟矩阵链乘运算的方法，并通过结构体实现矩阵的存储与计算。该算法首先读取矩阵的维度，然后解析矩阵链乘表达式，利用栈进行模拟计算，最终输出矩阵乘法所需的总次数。

题目链接：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;
}

转载于:https://www.cnblogs.com/Jstyle-continue/p/6351973.html