442 - Matrix Chain Multiplication

最新推荐文章于 2022-07-13 22:21:57 发布

原创最新推荐文章于 2022-07-13 22:21:57 发布 · 525 阅读

0 ·

CC 4.0 BY-SA版权

c/c++ 同时被 2 个专栏收录

91 篇文章

订阅专栏

算法

31 篇文章

订阅专栏

本文介绍了一个使用 C++ 实现的矩阵乘法表达式解析程序。该程序通过读取输入的矩阵乘法表达式，解析并计算出矩阵乘法过程中涉及的所有中间结果的累积代价。采用栈来管理矩阵操作，确保了表达式的正确处理。

#include<iostream>
#include<stack>
#include<string>
using namespace std;

struct Matrix{
  int a,b;
  Matrix(int a=0,int b=0):a(a),b(b){}
} mats[26];

int main()
{
  int N;
  cin>>N;
  for(int i=0;i<N;i++)
  {
    char num;
    cin>>num;
    int k=num-'A';
    cin>>mats[k].a>>mats[k].b;
  }
  string expr;
  while(cin>>expr)
  {
    stack<Matrix> s;
    int i=0,N=expr.size(),mul=0;
    bool ok=true;
    while(i<N)
    {
      if(expr[i]!='(')
      {
        if(expr[i]==')')
        {
          Matrix y=s.top();
          s.pop();
          Matrix x=s.top();
          s.pop();
          if(x.b!=y.a){cout<<"error"<<endl;ok=false;break;}
          mul+=x.a*x.b*y.b;
          s.push(Matrix(x.a,y.b));
        }
        else
          s.push(mats[expr[i]-'A']);
      }
      i++;
    }
    if(ok)cout<<mul<<endl;
  }
  return 0;
}