题目要求
3. 编写程序:输入表达式,出相应二叉树的先遍历结果
输入: a+b*(c -d) -e/f
输出: -+a*b -cd/ef
解题思路
一个编程题没有思路是没有任何解决办法的,所以正要的是培养解题的思路。
我素来很讨厌这样的题目,总感觉有种胆怯在里面。这次看了刘汝佳的算法竞赛入门经典的表达式树,感觉真的是绝妙极了。
下次可以深入的思考一下,如果换成struct或者用class来表示结点呢?
中序遍历结果呢,后序遍历呢?
......
代码如下:
主要代码是书本上的
#include<iostream>
#include<string>
using namespace std;
const int maxn = 1000;
int lch[maxn],rch[maxn]; //每个节点的左右孩子编号
char op[maxn]; //节点的字符
int cnt = 0; //节点数
//s的[x,y)作为范围
int build_tree(char *s,int x,int y){
int c1=-1,c2=-1,p=0,u;//c1,c2分别记录出现在括号外的最右边的加减号和乘除号
//当前表达式的长度为1,则直接为一棵树
if(y-x==1){
u = cnt++; //第u个节点
lch[u]=rch[u]=-1;
op[u]=s[x];
return u;
}
//p==0表示在括号外
for(int i=x;i<y;i++){
switch(s[i]){
case '(':p++;break;
case ')':p--;break;
case '+':
case '-':
if(p==0) c1 = i;
break;
case '*':
case '/':
if(p==0) c2 = i;
break;
}
}
if(c1<0) c1 = c2; //括号外没有+或者-,则选择括号外的*或/
if(c1<0) return build_tree(s,x+1,y-1); //括号外也没有*或/,说明表达式被一个括号括住,去括号
u=cnt++;
lch[u]=build_tree(s,x,c1);
rch[u]=build_tree(s,c1+1,y);
op[u]=s[c1];
return u;
}
void display(int a){
for(int i=0;i<cnt;i++){
cout<<op[i]<<" ";
}
}
int length(char *s){
int i=0;
while(s[i]!='\0'){
i++;
}
return i;
}
int main(){
char s[maxn];
cin>>s;
//cout<<" "<<length(s);
display(build_tree(s,0,length(s)));
cout<<endl;
return 0;
}