#include <iostream>
using namespace std;
// BST的结点
typedef struct node
{
int key;
struct node *lChild, *rChild;
}Node, *BST;
// 在给定的BST插入element, 使之称为新的BST
bool BSTInsert(Node * &p, int element)
{
if(NULL == p) // 空树
{
p = new Node;
p->key = element;
p->lChild = p->rChild = NULL;
return true;
}
if(element == p->key) // BST中不能有相等的值
return false;
if(element < p->key) // 递归
return BSTInsert(p->lChild, element);
return BSTInsert(p->rChild, element); // 递归
}
// 建立BST
void createBST(Node * &T, int a[], int n)
{
T = NULL;
int i;
for(i = 0; i < n; i++)
{
BSTInsert(T, a[i]);
}
}
// BST的查找(非递归)
Node *BSTSearch(BST T, int x)
{
Node *p = T;
while(NULL != p && x != p->key) // 找不到,就一直找
{
if(x > p->key)
p = p->rChild;
else
p = p->lChild;
}
return p;
}
int main()
{
int a[10] = {4, 5, 2, 1, 0, 9, 3, 7, 6, 8};
int n = 10;
BST T = NULL;
// 并非所有的a[]都能构造出BST,所以,最好对createBST的返回值进行判断
createBST(T, a, n);
Node *p = NULL;
int x; // 待查找的x
for(x = -10; x < 20; x++)
{
p = BSTSearch(T, x);
cout << x << ":";
if(NULL == p)
cout << "no" << endl;
else
cout << "yes" << endl;
}
return 0;
}
结果为:
-10:no
-9:no
-8:no
-7:no
-6:no
-5:no
-4:no
-3:no
-2:no
-1:no
0:yes
1:yes
2:yes
3:yes
4:yes
5:yes
6:yes
7:yes
8:yes
9:yes
10:no
11:no
12:no
13:no
14:no
15:no
16:no
17:no
18:no
19:no