当前位置:首页 > 二叉树应用解决计算表达式的问题

二叉树应用解决计算表达式的问题

点击次数:1895  更新日期:2011-12-13
 昨天晚上,我花了大把的时间研究里面二叉树应用解决计算表达式的问题,一直就没理解,主要是觉得是不是自己错了,又懒,不愿意自己把代码敲到电脑里看看,结果浪费了很多时间。所以还是提醒大家,代码这种东西,有什么好多看的,觉得他错了就自己敲到电脑里去看看!其实也没错太多,就是少了一些东西,导致原代码里的括号完全没有意义,也就是说,书中的代码虽然考虑到了计算表达式中的括号,却什么都没有做,而这其实只要稍稍改进:加一个flag存储上次读到的char,如果是‘)’的话,就要把左式当成运算数来计算。
  好了,把正确的代码贴在下面:
  #include <iostream>
  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<<exp.result()<<endl;
  return 0;
  }