矩阵链乘 Matrix Chain Multiplication Uva442

题目描述:
输入n个矩阵的维度和一些矩阵链乘表达式,输出乘法的次数。如果乘法无法进行,输出error。假定A是m*n矩阵,B是n*p矩阵,那么AB是m*p矩阵,乘法次数为m*n*p。如果A的列数和B的行数不同,则乘法无法进行。
例如,A是50*10,B是10*20的,C是20*5的,则(A(BC))的乘法次数是10*20*5(BC的乘法次数)+50*10*5(A(BC)的乘法次数)=3500
输入:首先输入矩阵数量n,接下来n行输入矩阵的字母及其第一二维的基数,接下来一行输入需要计算的表达式。
输出:对于每一个表达式有一个输出,即需要乘法的次数,若乘法无法进行则输出error

样例输入输出
在这里插入图片描述
代码源自算法竞赛入门经典(刘汝佳编著)P142
本文只是增添一些注释
代码如下:

//6-3 矩阵链乘

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

struct Matrix{
    int a,b;
    Matrix(int a=0,int b=0):a(a),b(b){};//构造函数

}m[26]; //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].a>>m[k].b;
   }

   string expr;//保存需要计算的式子

   while(cin>>expr){
       int len=expr.length();
       bool _judge=true; //判断这个式子能否计算
       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.b==m2.a){           //矩阵相乘的前提是左矩阵的列数等于右矩阵的行数
                   ans+=m1.a*m1.b*m2.b;
                   s.push(Matrix(m1.a,m2.b));
               }
               else{
                   _judge=false;
                   break;
               }
           }

       }
       if(_judge)
           printf("%d\n",ans);
       else {
           printf("error\n");
       }
   }
   return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值