6-12 二叉搜索树的操作集 (30分)
6-12 二叉搜索树的操作集 (30分)
本题要求实现给定二叉搜索树的5种常用操作。
函数接口定义:
BinTree Insert( BinTree BST, ElementType X );
BinTree Delete( BinTree BST, ElementType X );
Position Find( BinTree BST, ElementType X );
Position FindMin( BinTree BST );
Position FindMax( BinTree BST );
其中BinTree结构定义如下:
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
ElementType Data;
BinTree Left;
BinTree Right;
};
函数Insert将X插入二叉搜索树BST并返回结果树的根结点指针;
函数Delete将X从二叉搜索树BST中删除,并返回结果树的根结点指针;如果X不在树中,则打印一行Not Found并返回原树的根结点指针;
函数Find在二叉搜索树BST中找到X,返回该结点的指针;如果找不到则返回空指针;
函数FindMin返回二叉搜索树BST中最小元结点的指针;
函数FindMax返回二叉搜索树BST中最大元结点的指针。
裁判测试程序样例:
#include <stdio.h>
#include <stdlib.h>
typedef int ElementType;
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
ElementType Data;
BinTree Left;
BinTree Right;
};
void PreorderTraversal( BinTree BT ); /* 先序遍历,由裁判实现,细节不表 */
void InorderTraversal( BinTree BT ); /* 中序遍历,由裁判实现,细节不表 */
BinTree Insert( BinTree BST, ElementType X );
BinTree Delete( BinTree BST, ElementType X );
Position Find( BinTree BST, ElementType X );
Position FindMin( BinTree BST );
Position FindMax( BinTree BST );
int main()
{
BinTree BST, MinP, MaxP, Tmp;
ElementType X;
int N, i;
BST = NULL;
scanf("%d", &N);
for ( i=0; i<N; i++ ) {
scanf("%d", &X);
BST = Insert(BST, X);
}
printf("Preorder:"); PreorderTraversal(BST); printf("\n");
MinP = FindMin(BST);
MaxP = FindMax(BST);
scanf("%d", &N);
for( i=0; i<N; i++ ) {
scanf("%d", &X);
Tmp = Find(BST, X);
if (Tmp == NULL) printf("%d is not found\n", X);
else {
printf("%d is found\n", Tmp->Data);
if (Tmp==MinP) printf("%d is the smallest key\n", Tmp->Data);
if (Tmp==MaxP) printf("%d is the largest key\n", Tmp->Data);
}
}
scanf("%d", &N);
for( i=0; i<N; i++ ) {
scanf("%d", &X);
BST = Delete(BST, X);
}
printf("Inorder:"); InorderTraversal(BST); printf("\n");
return 0;
}
/* 你的代码将被嵌在这里 */
思路:二叉搜索树分为插入,查找,查找最大最小以及删除操作
插入操作,如果为空就生成一个结点,不然就比较插入值大小,如果小,往左递归比较,如果大,往右递归比较,查找同理,比较大小,小往左查找,大就往右查找,同理,最小最大就分别找左边叶结点和右边叶结点,
删除操作,首先查找待删除结点位置,如果是叶结点,直接free掉即可,如果有一个子结点,将子节点代替删除结点,如果删除结点位置有两个子结点
从右子树找最小的填充到删除位置,保证搜索树的平衡,再删除拿来填充的那个结点
BinTree Insert( BinTree BST, ElementType X ){
//如果是一个空节点
if(!BST){
BST = (BinTree)malloc(sizeof(struct TNode));//既然为空所以要生成一个
BST->Data = X;
BST->Left = NULL;
BST->Right = NULL;
}
else{//一般情况
if(X < BST->Data){//插入值小于节点,应该往左子树中找位置
BST->Left = Insert(BST->Left,X);//递归插入左子树
}
else if(X > BST->Data){//插入值大于节点,应该往右子树中找
BST->Right = Insert(BST->Right,X);//递归插入右子树
}
//如果相等说明X已经存在,什么也不做
}
return BST;
}
Position Find( BinTree BST, ElementType X ){
while(BST){//直接循环查找,类似链表
if(X < BST->Data){
BST = BST->Left;//小于节点,找左子树
}
else if(X > BST->Data){//大于节点,找右子树
BST = BST->Right;
}
else{//相等则找到
return BST;
}
}
return NULL;
}
Position FindMin( BinTree BST ){
if(!BST){
return NULL;
}
else if(!BST->Left)
return BST;
else return FindMin(BST->Left);
}
Position FindMax( BinTree BST ){
if(!BST)return NULL;
else if(!BST->Right)return BST;
else return FindMax(BST->Right);
}
BinTree Delete( BinTree BST, ElementType X ){
Position temp;
if(!BST){
printf("Not Found\n");//如果最终树为空,说明没有
}
else{//这里类似于插入重点在于找到后怎么办
if(X < BST->Data){
BST->Left = Delete(BST->Left,X);//从左子树递归删除
}
else if(X > BST->Data){
BST->Right = Delete(BST->Right,X);//从右子树递归删除
}
else{//当前BST就是要删除的节点
if(BST->Left && BST->Right){//要被删除的节点有左右两个孩子,就从右子树中找最小的数填充删除的节点
temp = FindMin(BST->Right);//找最小
BST->Data = temp->Data;//填充删除的节点
BST->Right = Delete(BST->Right,temp->Data);//删除拿来填充的那个节点
}
else{//只有一个子节点
temp = BST;
if(!BST->Left){//只有右节点
BST = BST->Right;//直接赋值就可以
}
else if(!BST->Right){//只有左节点
BST = BST->Left;//直接赋值就可以
}
free(temp);//如果啥也没有直接删除就可以,当然上面两种情况赋值后也要删除
}
}
}
return BST;
}