本题要求实现给定二叉搜索树的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;
}
/* 你的代码将被嵌在这里 */
输入样例:
10
5 8 6 2 4 1 0 10 9 7
5
6 3 10 0 5
5
5 7 0 10 3
输出样例:
Preorder: 5 2 1 0 4 8 6 7 10 9
6 is found
3 is not found
10 is found
10 is the largest key
0 is found
0 is the smallest key
5 is found
Not Found
Inorder: 1 2 4 6 8 9
源代码:
BinTree Insert( BinTree BST, ElementType X ){
if(BST==NULL){
BST=(BinTree)malloc(sizeof(BinTree));
BST->Data=X;
BST->Right=NULL;
BST->Left=NULL;
}
else{
if(X>BST->Data)
BST->Right=Insert(BST->Right,X);
else
BST->Left=Insert(BST->Left,X);
}
return BST;
}
BinTree Delete( BinTree BST, ElementType X ){
Position t;
if(!BST)
printf("Not Found\n");
else if(BST->Data>X)
BST->Left=Delete(BST->Left,X);
else if(BST->Data<X)
BST->Right=Delete(BST->Right,X);
else if(BST->Data==X){
if(BST->Left&&BST->Right){
t=FindMin(BST->Right);
BST->Data=t->Data;
BST->Right=Delete(BST->Right,BST->Data);
}
else{
t=BST;
if(!BST->Left)
BST=BST->Right;
else if(!BST->Right)
BST=BST->Left;
free(t);
}
}
return BST;
}
Position Find( BinTree BST, ElementType X ){
if(!BST)
return NULL;
else{
if(BST->Data==X)
return BST;
else if(BST->Data>X)
BST=Find(BST->Left,X);
else
BST=Find(BST->Right,X);
}
return BST;
}
Position FindMin( BinTree BST ){
if(BST){
while(BST->Left){
BST=BST->Left;
}
}
return BST;
}
Position FindMax( BinTree BST ){
if(BST){
while(BST->Right){
BST=BST->Right;
}
}
return BST;
}
总结:本题的Delete函数,我一开始想额外定义一个指针,指向删除结点的父节点,但是后来观看网上代码时发现完全不用。如果
①被删除的结点是叶子结点:直接删去该结点。
②被删除的结点只有左子树或者只有右子树,直接用左子树或者右子树替换它。
③被删除的结点既有左子树,也有右子树,则找到右子树中最小的结点,来替换删除结点;若替换删除结点仍有左右子树,则递归依次替换。
疑问:一开始在pta提交时一直出错,转到vs后运行正确无误,后来与网上代码进行比较,发现是删除结点只有左子树或右子树时的代码出现问题,如下是pta上出现问题代码
if(BST->Right)
BST=BST->Right;
else if(BST->Left)
BST=BST->Left;
pta结果:Inorder: 4 1791538064 1 2 4 6 4 1791537840 8 9
vs结果:
后来换成:,则pta上正确,vs也正确,我寻思两种方式不都是一个意思,为什么pta上结果不同,这个问题花了我快半个下午,麻了麻了。
希望有大佬解惑。