【栈 && 矩阵的知识】UVA 442 Matrix Chain Multiplication

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

翻译最新推荐文章于 2022-11-21 14:59:18 发布 · 275 阅读

文章标签：

#栈

UVa 紫书同时被 2 个专栏收录

38 篇文章

订阅专栏

栈和队列

15 篇文章

订阅专栏

本文介绍了一种通过解析输入字符串来计算多个矩阵相乘次数的方法。利用栈结构存储矩阵维度，并检查矩阵是否可以相乘。当遇到闭合括号时，取出栈顶两个矩阵检查它们的相乘条件并计算乘法次数。

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

Problem Description

给你n个矩阵，每个矩阵由一个字母命名，后面跟着矩阵的n行m列。
输入字符串要你求它相乘的次数，如果不能相乘输出error

代码：A的列数不等于B的行数不能相乘，假定A是m*n矩阵, B是n*p矩阵，那么AB是m*p矩阵，乘法次数m*n*p..

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int a, b;
};
node m[26];
stack<node> q;//存对应的值
int main()
{
    int n, k, i, len, ans;
    while(cin >> n)
    {
        string name;
        for(i = 0; i < n; i++)//先把对应字母的矩阵用结构体存起来
        {
            cin >> name;
            k = name[0] - 'A';
            cin >> m[k].a >> m[k].b;
        }
        string s;
        node x1, x2;
        while(cin >> s)
        {
            len = s.length();
            ans = 0;
            for(i = 0; i < len; i++)
            {
                if(s[i] == ')')//出来两个
                {
                    x2 = q.top(); q.pop();
                    x1 = q.top(); q.pop();
                    if(x1.b != x2.a) {printf("error\n"); break;}//不满足矩阵相乘
                    else
                    {
                        ans += x1.a * x1.b * x2.b;//满足，求乘的次数
                        q.push((node){x1.a, x2.b});//存起来
                    }
                }
                else if(s[i] >= 'A' && s[i] <= 'Z')//存起来
                q.push((node){m[s[i]-'A'].a, m[s[i]-'A'].b});
            }
            if(i == len) printf("%d\n", ans);
            while(!q.empty()) q.pop();
        }
    }
    return 0;
}