Implement Binary Search Tree

1. //BSTree.h
3. #include<iostream>
4. using namespace std;
5. template<typename Elem>
6. struct BSTNode     //implement BST's node
7. {
8.     BSTNode *left,*right;
9.     Elem data;
10.     BSTNode(Elem dataVal,BSTNode* leftVal=0,BSTNode* rightVal=0)
11.     {
12.         data = dataVal, left = leftVal, right = rightVal;
13.     }
14.     BSTNode(BSTNode* leftVal=0,BSTNode* rightVal=0)
15.     {
16.         left = leftVal, right = rightVal;
17.     }
18.     BSTNode *GetLeft(){return left;}
19.     BSTNode *GetRight(){return right;}
20.     Elem GetData(){return data;}
21.     void SetData(Elem it){data = it;}
22.     void SetLeft(BSTNode<Elem>*l){left = l;}
23.     void SetRight(BSTNode<Elem>*r){right = r;}
24. };
26. template<typename Elem>
27. class BSTree       //implement BST
28. {
29.     BSTNode<Elem>* root;
30. public:
31.     BSTree(Elem it){Init(it);}
32.     ~BSTree(){DeleteTree(root);}
33.     void Init(Elem);
34.     void InsertNode(BSTNode<Elem>*,Elem Item);
35.     void DeleteTree(BSTNode<Elem>*);
36.     void PreOrder(BSTNode<Elem>*);
37.     void InOrder(BSTNode<Elem>*);
38.     void PostOrder(BSTNode<Elem>*);
39.     bool Find(BSTNode<Elem>*,Elem);
40.     BSTNode<Elem>* GetRoot(){return root;}
41. };
42. //////////////////////////////////////////////////////////////////////
44. //BSTree.cpp
46. #include "BSTree.h"
47. template<typename Elem>
48. void BSTree<Elem>::Init(Elem item)
49. {
50.     root = new BSTNode<Elem>(item,0,0);
51. }
53. template<typename Elem>
54. void BSTree<Elem>::DeleteTree(BSTNode<Elem>* T)
55. {
56.     if (T != NULL)
57.     {
58.         DeleteTree(T->GetLeft());
59.         DeleteTree(T->GetRight());
60.         delete T;
61.     }
62. }
64. template<typename Elem>
65. void BSTree<Elem>::InsertNode(BSTNode<Elem>* T,Elem item)
66. {
67.     BSTNode<Elem>* p = 0;
68.     if (T == NULL)
69.     {
70.         Init(item);return ;
71.     }
72.     else p = T;
73.     while (1)
74.     {
75.         if (p->GetData() == item) return ;
76.         else if (p->GetData() > item)
77.         {
78.             if (p->GetLeft() == NULL)
79.             {
80.                 p->left = new BSTNode<Elem>(item,0,0);
81.                 return ;
82.             }
83.             else p = p->left;
84.         }
85.         else
86.         {
87.             if (p->GetRight() == NULL)
88.             {
89.                 p->right = new BSTNode<Elem>(item,0,0);
90.                 return ;
91.             }
92.             else p = p->right;
93.         }
94.     }
95. }
97. template<typename Elem>
98. void BSTree<Elem>::PreOrder(BSTNode<Elem>* T)
99. {
100.     if (T != NULL)
101.     {
102.         cout<<T->GetData()<<" ";
103.         PreOrder(T->GetLeft());
104.         PreOrder(T->GetRight());
105.     }
106. }
108. template<typename Elem>
109. void BSTree<Elem>::InOrder(BSTNode<Elem>* T)
110. {
111.     if (T != NULL)
112.     {
113.         InOrder(T->left);
114.         cout<<T->data<<" ";
115.         InOrder(T->right);
116.     }
117. }
119. template<typename Elem>
120. void BSTree<Elem>::PostOrder(BSTNode<Elem>* T)
121. {
122.     if( T != NULL)
123.     {
124.         PostOrder(T->left);
125.         PostOrder(T->right);
126.         cout<<T->data<<" ";
127.     }
128. }
130. template<typename Elem>
131. bool BSTree<Elem>::Find(BSTNode<Elem>* T,Elem item)
132. {
133.     if (T == NULL) return false;
134.     else if (T->GetData() > item) return Find(T->left,item);
135.     else if (T->GetData() < item) return Find(T->right,item);
136.     else
137.     {
138.         item = T->GetData();
139.         return true;
140.     }
141. }
143. ////////////////////////////////////////////////////////////
145. //main.cpp
147. #include "BSTree.cpp"
148. int main(void)
149. {
150.     BSTree<char> BSt('a');  //root node
151.     BSTNode<char>* BSn;
152.     BSn = BSt.GetRoot();
153.     cout<<"Insert number in the tree,terminted when you type '*'"<<endl;
154.     char item;
155.     cin>>item;
156.     while (item != '*')
157.     {
158.         BSt.InsertNode(BSn,item);
159.         cin>>item;
160.     }
161.     cout<<"PreOrder output the tree:"<<endl;
162.     BSt.PreOrder(BSn);
163.     cout<<endl<<"Inorder output the tree:"<<endl;
164.     BSt.InOrder(BSn);
165.     cout<<endl<<"PostOrder output the tree:"<<endl;
166.     BSt.PostOrder(BSn);
167.     char it;
168.     cin>>it;
169.     if(BSt.Find(BSn,it)) cout<<"There have the some number in the tree"<<endl;
170.     else cout<<"No found "<<it<<" in the tree."<<endl;
171.     return 0;
172. }