UVA - 442 Matrix Chain Multiplication

这是一道经典且比较难的栈题
洛谷链接
**

题面介绍

**
在这里插入图片描述

输入格式

在这里插入图片描述

输出格式

在这里插入图片描述
知道你们看不懂的中文翻译:

矩阵链乘
题目描述
​假设你必须评估一种表达形如 ABCDE,其中 A,B,C,D,E是矩阵。既然矩阵乘法是关联的,那么乘法的顺序是任意的。然而,链乘的元素数量必须由你选择的赋值顺序决定。
​ 例如,A,B,C分别是 50 * 10 ,10 * 20 和 20 * 5 的矩阵。现在有两种方案计算 A * B * C ,即(A * B) * C 和 A*(B * C)。
第一个要进行15000次基本乘法,而第二个只进行3500次。
​ 你的任务就是写出一个程序判定以给定的方式相乘需要多少次基本乘法计算。
输入格式
​ 输入包含两个部分:矩阵和表达式。 输入文件的第一行包含了一个整数 n(1 \leq≤ n \leq≤ 26), 代表矩阵的个数。接下来的n行每一行都包含了一个大写字母,说明矩阵的名称,以及两个整数,说明行与列的个数。
第二个部分严格遵守以下的语法:
SecondPart = Line { Line } Line = Expression Expression = Matrix | “(” Expression Expression “)” Matrix = “A” | “B” | “C” | … | “X” | “Y” | “Z”
输出格式
对于每一个表达式,如果乘法无法进行就输出 " error "。否则就输出一行包含计算所需的乘法次数。

以上为原文翻译,然而原文有误:
在这里插入图片描述
本题要求掌握的知识点有:
对栈熟练的理解,和对矩阵相乘的了解。
本题的坑点在于原文相乘数据出错,让博主算了半天,最后还去翻宋浩老师的《线性代数》…,才确定题目数据出错。
在题目中A*B应消去中间重复的相同值,即:50 * 10 * 20=10000。同理 B * C = 10 * 20 * 5 =1000.
总结就是如果有两个矩阵相乘,且第一个矩阵的列数等于第二个矩阵的行数,相乘结果为:第一个矩阵的行数X第一个矩阵列数X第二个矩阵列数
本题第二个坑点在于栈的原理是后进先出
因此当匹配到要进行相乘操作的矩阵时,先弹出来的是后面的矩阵,后弹出的是前面的矩阵
矩阵之间没有乘法交换律,需要认清数据再匹配。
AC代码:

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <climits>
#include <iomanip>
#define Buff std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define inf LONG_LONG_MAX
#define Inf INT_MAX
#define MEM(x, y) memset(x, y, sizeof(x))
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define rep(j, a, b) for (int j = (a); j <= (b); j++)
#define rrep(i, a, b) for (int i = (b); i >= (a); i--)
using namespace std;
const int MX = 1e7 + 10;
const ll Mod = 1e9 + 7;
const int INF = -0x3f3f3f3f;
struct Matrix
{
  int a, b;
  Matrix(int a = 0, int b = 0) : a(a), b(b) {}//定义矩阵结构体
} m[34];
stack<Matrix> s;
int main()
{
  Buff;
  int n;
  char c;
  string str;
  cin >> n;
  for (int i = 0; i < n; i++)
  {
    cin >> c;
    int k = c - 'A';//矩阵从A开始先减去A变为整数
    cin >> m[k].a >> m[k].b;//输入矩阵的行列 
  }
  while (cin >> str)//输入与字符相乘的字符串
  {
    int len = str.length();
    bool flag = false;
    int ans = 0;//记录乘法次数
    for (int i = 0; i < len; i++)
    {
      if (isalpha(str[i]))//小技巧一个判断是否为字母的库函数
        s.push(m[str[i] - 'A']);//如果是字母立即入栈,并处理为节点入栈
      else if (str[i] == ')')//如果有括号代表要相乘
      {
        Matrix m2 = s.top();//先出来的后面的矩阵!!!!!这点很重要
        s.pop();
        Matrix m1 = s.top();//后出来的是前面的矩阵!!!!
        s.pop();
        if (m1.b != m2.a)
        {
          flag = true;
          break;
        }//行列数不匹配直接break
        ans += m1.a * m1.b * m2.b;//匹配记录次数
        s.push(Matrix(m1.a, m2.b));
      }
    }
    if (flag)
      cout << "error" << endl;
    else
      cout << ans << endl;
  }
  return 0;
}

感谢收看,希望对大家有帮助

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值