编写程序实现下面运算:在二叉排序树中查找关键字为key的记录。
输入:
从键盘按先序输入,直到输入关键字等于-1为止,程序会自动创建一棵二叉树;再输入要查找的元素
输出:
若元素不存在输出:没有找到!
若元素存在:成功! 这个元素是 …
存储结构:
链式存储
算法的基本思想:
先在建立一棵二叉排序树,元素从键盘按先序输入,直到输入关键字等于-1为止,在二叉排序树 bt 中查找其关键字等于给定值的结点是否存在,并输出相应信息
源程序
#include <bits/stdc++.h>
using namespace std;
#define NULL 0
#define EQ(a,b) ((a)==(b))
#define LT(a,b) ((a)<(b))
typedef int KeyType;
typedef struct
{
KeyType key;
}ElemType; //元素类型
//定义二叉树的构成元素
typedef struct BSTNode
{
ElemType data;
struct BSTNode* lchild, * rchild;
}BSTNode, * BSTree;
BSTree searchBST(BSTree bt, KeyType key)
{
/*在二叉排序树 bt 中查找其关键字等于给定值的结点是否存在,并输出相应信息*/
if (bt == NULL) return NULL;//在排序二叉树中进行递归查找
else if EQ(bt->data.key,key) return bt;
else if LT(key , bt->data.key) return searchBST(bt->lchild, key);
else return searchBST(bt->rchild, key);
}
void insertBST(BSTree* bt, BSTree s)
{
/*在二叉排序树中插入一个新结点,即依次插入输入的数*/
if (*bt == NULL) *bt = s;
else if (s->data.key < (*bt)->data.key) insertBST(&((*bt)->lchild), s);
else if (s->data.key > (*bt)->data.key) insertBST(&((*bt)->rchild), s);
}
int main()
{
system("color f0");
char ch;
KeyType key;
BSTree bt, s;
int i = 0;
/*建立一棵二叉排序树,元素从键盘按先序输入,直到输入关键字等于-1为止*/
printf("\n请输入元素(-1:结束):\n");//以-1为结束
scanf("%d", &key);
bt = NULL;
while (key != -1)
{
s = (BSTree)malloc(sizeof(BSTNode));
(s->data).key = key; s->lchild = s->rchild = NULL;
insertBST(&bt, s);
scanf("%d", &key);
}//while
/*二叉排序树的查找,并输出查找的结果*/
printf("\n输入你想要查找的元素:");
scanf("%d", &key);
s = searchBST(bt, key);
if (s != NULL) printf("\n成功! 这个元素是 %d.\n", s->data.key);
else printf("\n没有找到!\n");
system("pause");
return 0;
}
该程序完整地满足了题目的所有要求,空间上没有额外空间,性能较好。
时间复杂度:
O(logn)
空间复杂度:
O(1)
本题需要操作者自己创建一个已经排好序的二叉排序树,本例中使用先序排列的方式创建一个二叉排序树;
通过SearchBST(BiTree T, KeyType key)函数,可以返回一个指向key的指针,若指针为空,则说明二叉排序树中没有关键字,若不为空说明有关键字;