例题6-3 矩阵链乘 用栈求表达式的值
拿着算法竞赛入门经典第二版大半个学期了,一直也没看,哎,今天看了一道,记录一下咯,毕竟书也是借的不然看完等脑子里忘光就等于白看了…题也直接拍的 好懒哦我
用栈求表达式的值:
之前写计算器的时候还不知道栈是啥,就if else 几千行这么傻乎乎的码,到头来还得分低,真是书到用时方恨少啊,这里就碰见了
题目:矩阵链乘
算法竞赛入门经典第二版 刘汝佳 P141例题6-3
代码:
代码就是书上的原代码,只是加上了自己体会的注释(保佑我自己以后还看得懂)
#include <bits/stdc++.h>
using namespace std;
struct Matrix{ //变量a,b存储维度,即a*b
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';//ASCLL码 如果输入矩阵B则k=1,m[1]=B
cin>>m[k].a>>m[k].b;
}
string expr;
while(cin>>expr)//输入表达式
{
int len=expr.length();//.length()函数求字符串长度
bool error=false;
int ans=0;
for(int i=0;i<len;i++)
{
if(isalpha(expr[i]))//函数isalpha()用于判断是否是英文字母
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)//m1的列数不等于m2的行数乘法无法进行
{
error=true;break;
}
ans+=m1.a*m1.b*m2.b;//进行乘法
s.push(Matrix(m1.a,m2.b));//结果入栈
}
}
if(error)cout<<"error"<<endl;//printf("error\n");
else cout<<ans<<endl;//printf("%d\n",ans);
}
return 0;
}
认识了一个新的小伙伴: