博客

#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