例题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;
}

认识了一个新的小伙伴:
在这里插入图片描述

运行一下咯:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_42324771/article/details/87094178
今日推荐