当前位置: 代码迷 >> 综合 >> UVA-442 Matrix Chain Multiplication【栈+模拟】
  详细解决方案

UVA-442 Matrix Chain Multiplication【栈+模拟】

热度:99   发布时间:2023-11-20 06:06:54.0

题意:输入n个矩阵的维度和一些矩阵链乘表达式;输出乘法的次数。如果两个矩阵无法相乘,输出error。

题解:列:矩阵A维度 m*n,B 维度 n*p,则 AB 维度 m*p, 乘法的次数 m*n*p;

出现error的情况是 A的列不等于B的行;我们可以用栈 存储矩阵链乘表达式的字母,遇到右括号出栈计算,并将计算的结果入栈;

#include<iostream>
#include<cstring>
#include<stack>
using namespace std;
struct Matrix{int x,y;Matrix(int x = 0, int y = 0):x(x),y(y) {}//构造函数
}m[26];
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].x >> m[k].y;}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.y != m2.x) { error = true; break;}ans += m1.x * m1.y * m2.y;s.push(Matrix(m1.x, m2.y));//将结果入栈}}if(error) printf("error\n");else printf("%d\n", ans);}return 0;
}

  相关解决方案