用栈实现的一个怪怪的二叉树前序遍历

书上的看不懂 网上的又懒的看 .......就这脾气
我判断栈的平衡是用入栈出栈的次数 - - 
怪就怪在这 2次入栈2次出栈后再出栈
记录节点的结构体变量还为真 万不得已才这样..... 
高手勿视  害眼自负  菜鸟一块来讨论学习下吧
还有代码过敏者勿视

 

// BiTree.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #define MAX_TREE_SIZE 100 #define STACK_INIT_SIZE 100 #define STACK_INCREMENT 10 typedef struct BiTNode{ int date; struct BiTNode *lchild,*rchild; }BiTNode,*BiTree; // 记录树节点 typedef struct{ BiTree *top; BiTree *base; int stacksize; }SqStack; //堆栈 BiTree T; SqStack S; //初始化栈 bool InitStack(SqStack &s){ s.base=(BiTree *)malloc(STACK_INIT_SIZE * sizeof(BiTree)); s.top=s.base; s.stacksize=STACK_INIT_SIZE; return true; } // 将结构体变量压栈 bool Push(SqStack &s,BiTree e){ if(s.top-s.base>=s.stacksize){ s.base=(BiTree *)realloc(s.base,(s.stacksize+STACK_INCREMENT) * sizeof(BiTree)); s.top=s.base+s.stacksize; s.stacksize+=STACK_INCREMENT; } *(s.top)=e; s.top++; return true; } //结构体变量出栈 BiTree Pop(SqStack &s,BiTree &e){ s.top--; e=*(s.top); return e; } //创造一棵树 bool CreatBiTree(BiTree &T){ int ch=0; std::cin>>ch; if(ch==0) T=NULL; else{ T=(BiTNode *)malloc(sizeof(BiTNode)); T->date=ch; CreatBiTree(T->lchild); CreatBiTree(T->rchild); } return true; } //二叉树前序遍历 bool PreOrderTraverse(SqStack S,BiTree T){ BiTree P=T; int push_count=0,pop_count=0; while (pop_count<=push_count) //没平衡栈就循环 { if (T->lchild) //左枝不为空就向左走 { Push(S,T); push_count++; std::cout<<T->date; T=T->lchild; } else if (T->rchild) //左枝为空就向右走 { Push(S,T); push_count++; std::cout<<T->date; T=T->rchild; } else //无路可走了就顺着来的方向返回 { std::cout<<T->date; P=T; T=Pop(S,T); pop_count++; while(pop_count<=push_count){ //栈不平衡 if(T->lchild==P){ if(T->rchild){ Push(S,T); push_count++; T=T->rchild; break; } } P=T; T=Pop(S,T); pop_count++; } } } return true; } int _tmain(int argc, _TCHAR* argv[]) { InitStack(S); CreatBiTree(T); PreOrderTraverse(S,T); return 0; }