基于二叉树的表达式求值算法

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
char s[N];
bool judge(char c){        //判断是不是数字
	return c>='0'&&c<='9';
}
struct node{
	char c;
	int num;
	node* left;
	node* right;
};
typedef node* tree;
tree build(int l,int r){          //建树
	tree p=new node;
	p->c='#';
	p->left=NULL;
	p->right=NULL;
	int a=0;
	int i;
	bool flag=false;
	while(l<r&&s[l]=='('&&s[r]==')') l++,r--;//略去无意义括号
	vector<int> v;//存储*/符号下标
	for(i=l;i<=r;i++){
		if(!judge(s[i])&&s[i]!='('&&s[i]!=')'&&a==0){  //括号外是符号且不是括号
			if(s[i]=='+'||s[i]=='-'){      //是+或-,则以它为父节点,建左右子树
				flag=true;
				p->c=s[i];
				p->left=build(l,i-1);
				p->right=build(i+1,r);
				break;
			}
			else{
				v.push_back(i); //存储*/下标
			}	
		}
		else if(s[i]=='(') a++;
		else if(s[i]==')') a--;
	}	
	if(!flag&&v.size()>0){ //括号外没有+-,则以最后一个符号为父节点
	    int m=v.size()-1;  
		p->c=s[v[m]];
		p->left=build(l,v[m]-1);
		p->right=build(v[m]+1,r);
	}
	if(l==r) {     //如果l==r,说明是一个数
	   p->c=s[l];
	   p->num=s[l]-'0';
	};
	return p;
} 
void print(tree t){     //没用
	if(t->left!=NULL) print(t->left);
	if(t->right!=NULL) print(t->right);
	if(t->left==NULL&&t->right==NULL) printf("%c",t->c);
}
int js(tree t){   //计算
	if(t->left!=NULL&&t->right!=NULL){    //左右子树都不为空,说明该节点是个符号,需要运算
		if(t->c=='*') t->num=js(t->left)*js(t->right);
		else if(t->c=='/') t->num=js(t->left)/js(t->right);
		else if(t->c=='+') t->num=js(t->left)+js(t->right);
		else if(t->c=='-') t->num=js(t->left)-js(t->right);
	}
    return t->num;//返回节点的值
}
int main(){
	while(~scanf("%s",s)){
		if(s[0]=='=') break;
		int len=strlen(s);
		int l=0,r=len-2; 
	    tree T=build(l,r);
	    int t=js(T);
	    cout<<t<<endl;
	}
}

 

全部评论

相关推荐

粗心的雪碧不放弃:纯学历问题,我这几个月也是一直优化自己的简历,后来发现优化到我自己都觉得牛逼的时候,发现面试数量也没有提升,真就纯学历问题
点赞 评论 收藏
分享
死在JAVA的王小美:哈哈哈哈哈哈哈哈哈哈哈哈哈哈哈哈哈哈,我也是,让我免了一轮,但是硬气拒绝了
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务