UVA - 442 C - Matrix Chain Multiplication

最新推荐文章于 2021-10-23 21:53:57 发布

原创最新推荐文章于 2021-10-23 21:53:57 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

紫书专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种使用栈和映射来优化矩阵乘法运算次数的方法，并通过C++实现了一个具体的例子，展示了如何处理矩阵表达式并计算其所需的最少乘法次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目 : 计算一个矩阵乘法的次数

讲道理，应该用struct的，但是最近几天写stl写傻了。。。直接上了pair，也不知道是为什么。。。

#include <algorithm>
#include <cstring>
#include <iostream>
#include <cstdio>
#include <queue>
#include <stack>
#include <map>
using namespace std;
#define Pair pair<int,int>
int n;
map<int ,pair<int,int> > mat;

int getdata()
{
    char s[1005];
    stack<pair<Pair,int> > S;
    while(cin.getline(s,1005))
    {
        int slen = strlen(s);
        int flag = 1;
        for(int i = 0; i < slen; i++)
        {
            if(s[i] == ')')
            {
                pair<Pair,int> b = S.top(); S.pop();
                pair<Pair,int> a = S.top(); S.pop();
                if(a.first.second != b.first.first)
                {
                    flag = 0;
                    printf("error\n"); break;
                }
                else
                {
                    int jieguo = (a.first.first)*(a.first.second)*(b.first.second);
                    S.push(pair<Pair,int>(Pair(a.first.first,b.first.second),jieguo+a.second+b.second));
                }
            }
            if(isupper(s[i]))
            {
                int a = s[i] - 'a';
                S.push(pair<Pair,int>(Pair(mat[a].first,mat[a].second),0));
            }
        }
        if(flag)
            printf("%d\n",S.top().second);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i = 0; i < n ; i++)
    {
        char s[2];
        int a,b;
        scanf("%s%d%d",s,&a,&b);
        mat[s[0] -'a'] = Pair(a,b);
    }
    getchar();
    getdata();
    return 0;
}