昨天晚上,我花了大把的时间研究里面二叉树应用解决计算表达式的问题,一直就没理解,主要是觉得是不是自己错了,又懒,不愿意自己把代码敲到电脑里看看,结果浪费了很多时间。所以还是提醒大家,代码这种东西,有什么好多看的,觉得他错了就自己敲到电脑里去看看!其实也没错太多,就是少了一些东西,导致原代码里的括号完全没有意义,也就是说,书中的代码虽然考虑到了计算表达式中的括号,却什么都没有做,而这其实只要稍稍改进:加一个flag存储上次读到的char,如果是‘)’的话,就要把左式当成运算数来计算。
好了,把正确的代码贴在下面:
#include using namespace std; class calc { enum Type {DATA, ADD, SUB, MULTI, DIV, OPAREN, CPAREN, EOL}; struct node { Type type; int data; node *lchild, *rchild; node(Type t, int d=0, node *lc=NULL, node *rc=NULL) { type=t; data=d; lchild=lc; rchild=rc; } }; node *root; node *create(char * &s); Type getToken (char * &s, int &value); int result (node *t); public: calc (char *s) {root=create(s);} int result() {if (root==NULL) return 0; return result(root);} }; calc::node *calc::create(char * &s) { node *p, *root=NULL; Type returnType,flag=DATA; int value; while (*s) { flag=returnType; returnType=getToken(s,value); switch (returnType) { case DATA: case OPAREN: if (returnType == DATA) p=new node(DATA,value); else p=create(s); if (root==NULL) root=p; else if (root->rchild==NULL) root->rchild=p; else root->rchild->rchild=p; break; case CPAREN: case EOL: return root; case ADD: case SUB: root=new node(returnType,0,root); break; case MULTI: case DIV: if (root->type==DATA || root->type==MULTI || root->type==DIV || flag==OPAREN) root=new node(returnType,0,root); else root->rchild=new node(returnType,0,root->rchild); } } return root; } calc::Type calc::getToken(char *&s, int &data) { char type; while (*s==' ') ++s; if (*s>='0' && *s<='9') { data=0; while (*s>='0' && *s<='9') {data=data*10+ *s-'0'; ++s;} return DATA; } if (*s == '0') return EOL; type =*s; ++s; switch(type) { case '+':return ADD; case '-':return SUB; case '*':return MULTI; case '/':return DIV; case '(':return OPAREN; case ')':return CPAREN; default: return EOL; } } int calc::result(node *t) { int num1,num2; if (t->type == DATA) return t->data; num1=result(t->lchild); num2=result(t->rchild); switch(t->type) { case ADD:t->data=num1+num2;break; case SUB:t->data=num1-num2;break; case MULTI: t->data=num1*num2;break; case DIV:t->data=num1/num2;break; } return t->data; } int main() { char equation[256]; cin>>equation; //calc exp("3*(2+(6/2))"); calc exp(equation); cout<
![4a6895bee93a69fe38c54be0c612e455.png](https://i-blog.csdnimg.cn/blog_migrate/6766a16ea9ad81d22f1571c962283600.jpeg)