二叉树的几种非递归建立与遍历的实现

1. #include<iostream>
2. #include<stdlib.h>
3. #include<stack>
4. #include<queue>
5. #include<string>
6. using namespace std;
8. #define OK 1
9. #define ERROR 0
10. #define OVERFLOW -2
11. typedef int Status;
12. typedef char ElemType;
13. typedef Status (*Visit)(ElemType e);
15. typedef struct  BiTNode
16. {
17.         ElemType data;
18.         struct BiTNode *lchild,*rchild;
19.         }BiTNode,*BiTree;
22. /*
23. Status CreateBiTree(BiTree &T,string s)  // 非递归按层次从左到右创建 
24. {
25.      queue<BiTree> q;
26.      BiTree p,m;
27.      if(s[0]=='#') T=NULL;
28.      else {
29.            T=new BiTNode;
30.            T->data=s[0];
31.            q.push(T); }
32.      int i=1;
33.      while(!q.empty())
34.       {
35.          p=q.front();
36.          q.pop();
37.          if(s[i]!='#') {
38.                         m=new BiTNode;
39.                         m->data=s[i++];
40.                         p->lchild=m;
41.                         q.push(m);
42.                         if(s[i]!='#') {
43.                                       m=new BiTNode;
44.                                       m->data=s[i++];
45.                                       p->rchild=m;
46.                                       q.push(m);}
47.                         else p->rchild=NULL;}
48.          else {
49.               p->lchild=NULL;
50.               if(s[i]=='#') 
51.      {p->rchild=NULL;i++;} }
52.          }
53.          return OK; }
55. */
58. Status CreateBiTree(BiTree &T,string str)   //非递归先序创建 
59. {
60.     int flag,k=0;
61.  size_t i=0;
62.     stack<BiTree> s;
63.     BiTree p,cur;
64.    if(str[0]=='#') {T=NULL;return OK;} 
65.    while(i<str.size())
66.    {
67.     if(str[i]!='#') {
68.          p=new BiTNode;
69.          p->data=str[i];
70.          s.push(p);
71.          flag=k?3:0;}
72.     else {flag++; i++;p=NULL;}    
73.     if(!i) {cur=T=p;i++;}
74.     else 
75.     switch(flag)
76.  {case 0:
77.              cur->lchild=p;
78.              cur=p;i++;
79.              break;
80.     case 1: 
81.             cur->lchild=p;
82.             break;
83.     case 2:
84.             cur->rchild=p;s.pop();
85.             if(i<str.size()) {
86.                              cur=s.top();
87.                              s.pop();}
88.             k=1;
89.             break;
90.    case 3: 
91.            cur->rchild=p;
92.            cur=p;
93.            k=0;
94.            i++;
95.            flag=0;
96.            break;};
97.     }return OK;
98. }
103. Status PrintElem(ElemType e)
104. {
105.    cout<<e;
106.    return OK; }
108. void PreOrderTraverse(BiTree T,Status(*Visit)(ElemType e)) {
109. //非递归先序遍历 
110. if(!T) return;
111. stack<BiTree> s;
112. BiTree p=T;
113. while(p||!s.empty())
114. {
115.    while(p)
116.        {
117.          Visit(p->data);
118.          s.push(p);
119.          p=p->lchild;  }
120.    p=s.top()->rchild;
121.    s.pop();
122.    }
123.  cout<<endl;}  
124. void InOrderTraverse(BiTree T,Status(*Visit)(ElemType e)) {
125. //非递归中序遍历 
126. stack<BiTree> m;
127. BiTree p=T;
128. while(p||!m.empty())   
129.  if(p) {m.push(p);p=p->lchild;}
130.  else {
131.   p=m.top();
132.   m.pop();
133.   Visit(p->data);
134.   p=p->rchild; } 
135. cout<<endl;}
137. void PostOrderTraverse(BiTree T,Status(*Visit)(ElemType e)) {
138. //非递归后序遍历
139. if(!T) return;
140. stack<BiTree> s;
141. BiTree p=T,pre=NULL;
142. while(p||!s.empty()){
143.    while(p)
144.        {
145.          s.push(p);
146.          p=p->lchild;  }
147.    p=s.top();
148.    if(!p->rchild||p->rchild==pre){
149.           Visit(s.top()->data);
150.           s.pop();
151.           pre=p;
152.           p=NULL;}     
153.    else  p=p->rchild;
154.    }
155. cout<<endl;}
159. Visit Vis=PrintElem;
161. int main()
162. {
163.    BiTree T;
164.    string s;
165.    cin>>s;
166.    CreateBiTree(T,s);
167.    PreOrderTraverse(T,Vis);
168.    InOrderTraverse(T,Vis);
169.    PostOrderTraverse(T,Vis);
170.    return 0; }