Educoder【实验4 基于哈夫曼树的数据压缩算法】【第11关：基于二叉树的表达式求值】

【第11关：基于二叉树的表达式求值】

任务描述

输入一个表达式（表达式中的数均为小于10的正整数），利用二叉树来表示该表达式，创建表达式树，然后利用二叉树的遍历操作求表达式的值。

编程要求

输入

多组数据。每组数据一行，为一个表达式，表达式以‘=’结尾。当输入只有一个“=”时，输入结束。

输出

每组数据输出一行，为表达式的值。

测试说明

测试输入：

2*(2+5)=

1+2=

=

预期输出：

14

3

代码部分

272.h：

​ #include<iostream>
#define MAXSIZE 100
using namespace std;
typedef struct BiTNode
{//二叉树的双链表存储表示
	double data;          //结点数据域
	bool ischaracter;      //判断结点是否为字符
	struct BiTNode *lchild,*rchild;    //左右孩子指针
}BiTNode,*BiTree;
typedef struct
{//字符栈的存储结构
	char *base;     //栈底指针
	char *top;       //栈顶指针
	int stacksize;   //栈可用的最大容量
}SqStack1;
typedef struct
{//结点栈的存储结构
	BiTree *base;
	BiTree *top;
	int stacksize;
}SqStack2;
void InitStack1(SqStack1 &s)
{//字符栈的初始化
	s.base = new char;
	if (!s.base)
	{
		cout << "内存分配失败!\n";
		exit(0);
	}
	s.top = s.base;
	s.stacksize = MAXSIZE;

}
void InitStack2(SqStack2 &s)
{//结点栈的初始化
	if (!(s.base = new BiTree))
		exit(0);
	s.top = s.base;
	s.stacksize = MAXSIZE;
}
void Push1(SqStack1 &s,char ch)
{//字符入栈操作
	if (s.top - s.base == s.stacksize) return;
	*s.top++ = ch;
}
void Push2(SqStack2 &s,BiTree t)
{//结点入栈操作
	if (s.top - s.base == s.stacksize) return;
	*s.top++ = t;

}
void Pop1(SqStack1 &s,char &ch)
{//字符出栈操作
	if (s.top == s.base)
		return ;
	ch = *--s.top;

}
void Pop2(SqStack2 &s,BiTree &t)
{//结点出栈操作
	if (s.top == s.base)
		return;
	t = *--s.top;

}
char GetTop(SqStack1 s)
{//取字符栈的栈顶元素
	if (s.top != s.base)
	return *(s.top - 1);
}
bool EmptyStack(SqStack1 s)
{//栈的判空操作
	if (s.top == s.base) return true;
	else return false;

}
char Precede(char a,char b)
{//判断符号的优先级
	int i, j;
	char pre[7][7] =
	{
			{'>','>','<','<','<','>','>'},
			{'>','>','<','<','<','>','>'},
			{'>','>','>','>','<','>','>'},
			{'>','>','>','>','<','>','>'},
			{'<','<','<','<','<','=','0'},
			{'>','>','>','>','0','>','>'},
			{'<','<','<','<','<','0','='}
	};
	switch (a) {
	case '+': i = 0; break;
	case '-': i = 1; break;
	case '*': i = 2; break;
	case '/': i = 3; break;
	case '(': i = 4; break;
	case ')': i = 5; break;
	case '=': i = 6; break;
	}
	switch (b) {
	case '+': j = 0; break;
	case '-': j = 1; break;
	case '*': j = 2; break;
	case '/': j = 3; break;
	case '(': j = 4; break;
	case ')': j = 5; break;
	case '=': j = 6; break;
	}
	return pre[i][j];
}
double Operate(double a,char ch,double b)
{//运算操作，返回相应的数值结果
	double c;
	switch (ch)
	{
	case '+':
		c = a + b;
		break;
	case '-':
		c = a - b;
		break;
	case '*':
		c = a * b;
		break;
	case '/':
		c = a / b;
		break;
	}
	return c;
}
bool IsCharacter(char ch)
{//判断ch是否为+、-、*、/、(、)、= 这类的字符，是则返回true
	switch (ch)
	{
	case '+':
	case '-':
	case '*':
	case '/':
	case '(':
	case ')':
	case '=':
		return true;
	default:
		return false;
	}

}
double InOrder(BiTree T)
{//中序遍历二叉树并求表达式的值
	double lvalue;
	double rvalue;
	char theta;
	if (T == NULL)
		return 0;
	else
	{
		if (T->lchild == NULL && T->rchild == NULL)
			return T->data - '0';
		else
		{
			lvalue = InOrder(T->lchild);
			theta = T->data;
			rvalue = InOrder(T->rchild);
			
			return Operate(lvalue, theta, rvalue);
		}
	}
}
void CreateBT(char ch[],BiTree &t,SqStack1 optr,SqStack2 expt)
{//创建二叉树
  char c,x, theta;
	BiTree a, b;
	int i = 0;
	c = ch[i++];
	x = GetTop(optr);
	while (c != '=' || x != '=')
	{
		if (IsCharacter(c))
		{
			switch (Precede(x, c))
			{
			case '<':
				Push1(optr, c);
				c = ch[i++];
				break;
 
			case '=':
				Pop1(optr,x);
				c = ch[i++];
				break;
			case '>':
				Pop1(optr, theta);
				BiTree TEMP;
				TEMP = new BiTNode;
				if (!TEMP)
					exit(-1);
				TEMP->data = theta;
				Pop2(expt, b);
				Pop2(expt, a);
 
				TEMP->lchild = a;
				TEMP->rchild = b;
				Push2(expt, TEMP);
 
				t = TEMP;
				break;
			}
		}
 
		if (c >= '0' && c <= '9')
		{
			BiTree Temp;
			Temp = new BiTNode;
			if (!Temp)
				exit(-1);
			Temp->data = c;
			Temp->lchild = NULL;
			Temp->rchild = NULL;
			Push2(expt, Temp);
 
			c = ch[i++];;
		}
		x = GetTop(optr);
 
	}
}

272.cpp 主函数文件(不可编辑）

 #include<iostream>
#include "272.h"
using namespace std;

int main()
{
	char ch[MAXSIZE];
	while(cin>>ch)
	{
		if(ch[0]=='=') break;
		BiTree t;
		SqStack1 optr;      //运算符栈optr
		SqStack2 expt;        //数字栈expt
		InitStack1(optr);     //初始化栈
		InitStack2(expt);     //初始化栈
		Push1(optr,'=');    //先在运算符栈底放入一个'='
		CreateBT(ch,t,optr,expt);       //创建二叉树
		double answer=InOrder(t);
		cout<<answer<<endl;
	}
	return 0;
}