二叉查找树

1. // BSTree.cpp : Defines the entry point for the console application.
2. //问题描述:二叉查找树的实现
3. //创建者:QGX
4. //创建时间:2008-2-1
5. #include "stdafx.h"
6. #include<iostream>
7. using namespace std;
9. //二叉查找树的定义
10. #define NULL 0
11. typedef  int Type;
13. class TreeNode
14. {
15.  friend class BSTree;
16. private:
17.  TreeNode *rchild,*lchild;
18.  Type data;
19. };
21. class BSTree
22. {
23. private:
24.  TreeNode *tree;
25.  TreeNode* Search(TreeNode *t,Type x);
26.  void traverse(TreeNode *t);
27. public:
28.  BSTree(){tree=NULL;}
29.  ~BSTree(){delete tree;}
30.  TreeNode* RSearch(Type x);
31.  TreeNode* ISearch(Type x);
32.  void Insert(Type x);
33.  void Delete(Type x);
34.  void Traverse(){traverse(tree);}
35. };
37. TreeNode* BSTree::RSearch(Type x)
38. {
39.  return Search(tree,x);
40. }
42. TreeNode* BSTree::Search(TreeNode* t,Type x)
43. {
44.  //递归查找
45.  if(!t)return NULL;
46.  if(t->data==x)return t;
47.  else if(t->data<x)return Search(t->rchild,x);
48.  else return Search(t->lchild,x);
49. }
51. TreeNode* BSTree:: ISearch(Type x)
52. {
53.  //迭代查找
54.  TreeNode* t=tree;
55.  bool found=false;
56.  while(!found && t)
57.  {
58.   if(t->data==x)
59.   {
60.    found=true;
61.   }
62.   else if(t->data<x)
63.   {
64.    t=t->rchild;
65.   }
66.   else t=t->lchild;
67.  }
68.  if(found) return t;
69.  else return NULL;
70. }
72. void BSTree:: Insert(Type x)
73. {
74.  TreeNode *p,*q;
75.  bool found=false;
76.  p=tree;
77.  q=NULL;
78.  while(p)
79.  {
80.   if(p->data==x)
81.   {
82.    found=true;
83.    break;
84.   }
85.   else
86.   {
87.    q=p;
88.    if(p->data>x)p=p->lchild;
89.    else p=p->rchild;
90.   }  
91.  }
92.  if(found)return;
93.  else
94.  {
95.   TreeNode *temp=new TreeNode;
96.   temp->lchild=NULL;
97.   temp->rchild=NULL;
98.   temp->data=x;
99.   
100.   if(!tree)tree=temp;//根结点
101.   else
102.   {
103.    if(q->data>x)q->lchild=temp;
104.    else q->rchild=temp;
105.   }
106.  }
107. }
109. void BSTree::Delete(Type x)
110. {
111.  TreeNode *p=tree,*q=tree;
112.  bool found=false;
113.  while(p)
114.  {
115.   if(p->data==x)
116.   { //找到待删除的结点
117.    found=true;
118.    break;
119.   }
120.   else
121.   { //q为p的父结点
122.    q=p;
123.    if(p->data>x)p=p->lchild;
124.    else p=p->rchild;
125.   }
126.  }
128.  if(!found)
129.  {
130.   //不存在此结点
131.   cout<<"Does not exist this node"<<endl;
132.         return;
133.  }
134.  else
135.  {
136.   if(p==q)
137.   {
138.    //删除的是根结点
139.    if(!p->rchild && !p->lchild)
140.    { //无左右子树
141.     delete p;
142.     tree=NULL;
143.    }
144.    else if(!p->rchild)
145.    { //无右子树 
146.     tree=p->lchild;
147.     delete p;
148.    }
149.    else if(!p->lchild)
150.    { //无左子树
151.     tree=p->rchild;
152.     delete p;
153.    }
154.    else
155.    {
156.     //有左右子树
157.     //找到x的直接后继结点,交换，删除
158.     TreeNode *p2=p->rchild;
159.     TreeNode *p3=p->rchild;
161.     while(p2->lchild)
162.     {
163.      p3=p2;
164.      p2=p2->lchild;
165.     }
166.     if(p3==p2)
167.     {
168.      p2->lchild=p->lchild;
169.      tree=p2;
170.      delete p;
171.     }
172.     else
173.     {
174.      p->data=p2->data;
175.      p3->lchild=p2->rchild;
176.      delete p2;
177.     }
178.    }
179.   }//end_if
181.   else
182.   {
183.    //删除非根结点
184.    if(!p->rchild && !p->lchild)
185.    {
186.     if(q->data>x)q->lchild=NULL;
187.     else q->rchild=NULL;
188.     delete p;
189.    }
190.    else if(!p->rchild)
191.    {
192.     if(q->data>x)q->lchild=p->lchild;
193.     else q->rchild=p->lchild;
194.     delete p;
195.    }
196.    else if(!p->lchild)
197.    {
198.     if(q->data>x)q->lchild=p->rchild;
199.     else q->rchild=p->rchild;
200.     delete p;
201.    }
202.    else
203.    {
204.     //有左右子树
205.     //找到x的直接后继结点,交换，删除
206.     TreeNode *p2,*p3;
207.     p2=p->lchild;
208.     p3=p->lchild;
209.     while(p2->rchild)
210.     {
211.      p3=p2;
212.      p2=p2->rchild;
213.     }
214.     if(p2==p3)
215.     {
216.      p->data=p2->data;
217.      p->lchild=p2->lchild;
218.      delete p2;
219.     }
220.     else
221.     {
222.      p->data=p2->data;
223.      p3->rchild=p2->lchild;
224.      delete p2;
225.     }
226.    }//end_else   
227.   }//end_else
228.  }//end_else
229. }
231. void BSTree::traverse(TreeNode *t)
232. {
233.  if(!t)return;
234.  else
235.  {
236.   traverse(t->lchild);
237.   cout<<t->data<<endl;
238.   traverse(t->rchild);
239.  }  
240. }
242. int _tmain(int argc, _TCHAR* argv[])
243. {
244.  BSTree *t=new BSTree();
245.  const int N=8;
246.  Type data[N]={8,6,5,7,12,9,20,14};
247.  for(int i=0;i<N;i++)
248.   t->Insert(data[i]);
249.  cout<<"The tree is:"<<endl;
250.  t->Traverse();
251.  //t->Delete(6); 
252.  //t->Delete(8);
253.  //t->Delete(12);
254.  //t->Delete(5);
255.  //t->Delete(20);
256.  //t->Insert(100);
257.  //t->Insert(3);
258.  //t->Insert(12);
259.  //t->Insert(19);
260.  //t->Traverse();
261.     //if(t->ISearch(19))cout<<"OK"<<endl;
262.  //if(t->RSearch(19))cout<<"OK too"<<endl;
263.  getchar();
264.  return 0;
265. }
266.