c++用二叉树表示代数表达式_C++二叉树应用：计算表达式

您可能感兴趣的话题：

C++

核心提示：昨天晚上，我花了大把的时间研究里面二叉树应用解决计算表达式的问题

昨天晚上，我花了大把的时间研究里面二叉树应用解决计算表达式的问题，一直就没理解，主要是觉得是不是自己错了，又懒，不愿意自己把代码敲到电脑里看看，结果浪费了很多时间。所以还是提醒大家，代码这种东西，有什么好多看的，觉得他错了就自己敲到电脑里去看看！其实也没错太多，就是少了一些东西，导致原代码里的括号完全没有意义，也就是说，书中的代码虽然考虑到了计算表达式中的括号，却什么都没有做，而这其实只要稍稍改进：加一个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<

return 0;

}