Update 9.二叉搜索树的操作.cpp

这里的free放错位置会出问题的,改写会使程序更简洁
This commit is contained in:
H1206950012
2020-02-13 15:13:29 +08:00
committed by GitHub
parent e32a4e3eb2
commit 5e9ff157e9

View File

@@ -9,100 +9,98 @@ struct TreeNode{
BinTree Right;
};
// 查找递归实现
// 查找递归实现
BinTree Find(ElementType X,BinTree BST){
if(!BST) // 如果根结点为空,返回 NULL
if(!BST) // 如果根结点为空,返回 NULL
return NULL;
if(X < BST->Data) // 比根结点小,去左子树查找
if(X < BST->Data) // 比根结点小,去左子树查找
return Find(X,BST->Left);
else if(BST->Data < X) // 比根结点大,去右子树查找
else if(BST->Data < X) // 比根结点大,去右子树查找
return Find(X,BST->Right);
else if(BST->Data == X) // 找到了
else if(BST->Data == X) // 找到了
return BST;
}
// 查找非递归实现
// 查找非递归实现
BinTree IterFind(ElementType X,BinTree BST){
while(BST){
if(X < BST->Data)
BST = BST->Left;
else if(BST->Data < X) // 比根结点大,去右子树查找
else if(BST->Data < X) // 比根结点大,去右子树查找
BST = BST->Right;
else if(BST->Data == X) // 找到了
else if(BST->Data == X) // 找到了
return BST;
}
return NULL;
}
// 查找最小值的递归实现
// 查找最小值的递归实现
BinTree FindMin(BinTree BST){
if(!BST) // 如果为空了,返回 NULL
if(!BST) // 如果为空了,返回 NULL
return NULL;
else if(BST->Left) // 还存在左子树,沿左分支继续查找
else if(BST->Left) // 还存在左子树,沿左分支继续查找
return FindMin(BST->Left);
else // 找到了
else // 找到了
return BST;
}
// 查找最大值的非递归实现
// 查找最大值的非递归实现
BinTree FindMax(BinTree BST){
if(BST) // 如果不空
while(BST->Right) // 只要右子树还存在
if(BST) // 如果不空
while(BST->Right) // 只要右子树还存在
BST = BST->Right;
return BST;
}
// 插入
// 插入
BinTree Insert(ElementType X,BinTree BST){
if(!BST){ // 如果为空,初始化该结点
if(!BST){ // 如果为空,初始化该结点
BST = (BinTree)malloc(sizeof(struct TreeNode));
BST->Data = X;
BST->Left = NULL;
BST->Right = NULL;
}else{ // 不为空
if(X < BST->Data) // 如果小,挂在左边
}else{ // 不为空
if(X < BST->Data) // 如果小,挂在左边
BST->Left = Insert(X,BST->Left);
else if(BST->Data < X) // 如果大,挂在右边
else if(BST->Data < X) // 如果大,挂在右边
BST->Right = Insert(X,BST->Right);
// 如果相等,什么都不用做
// 如果相等,什么都不用做
}
return BST;
}
// 删除
// 删除
BinTree Delete(ElementType X,BinTree BST){
BinTree tmp;
if(!BST)
cout<<"要删除的元素未找到";
else if(X < BST->Data) // X 比当前结点值小,在左子树继续查找删除
cout<<"要删除的元素未找到";
else if(X < BST->Data) // X 比当前结点值小,在左子树继续查找删除
BST->Left = Delete(X,BST->Left);
else if(BST->Data < X) // x 比当前结点值大,在右子树继续查找删除
else if(BST->Data < X) // x 比当前结点值大,在右子树继续查找删除
BST->Right = Delete(X,BST->Right);
else{ // 找到被删除结点
if(BST->Left && BST->Right){ // 被删除结点有俩孩子结点
tmp = FindMin(BST->Right); // 找到右子树中值最小的
BST->Data = tmp->Data; // 用找到的值覆盖当前结点
BST->Right = Delete(tmp->Data,BST->Right); // 把前面找到的右子树最小值结点删除
}else{ // 被删除结点只有一个孩子结点或没有孩子结点
else{ // 找到被删除结点
if(BST->Left && BST->Right){ // 被删除结点有俩孩子结点
tmp = FindMin(BST->Right); // 找到右子树中值最小的
BST->Data = tmp->Data; // 用找到的值覆盖当前结点
BST->Right = Delete(tmp->Data,BST->Right); // 把前面找到的右子树最小值结点删除
}else{ // 被删除结点只有一个孩子结点或没有孩子结点
tmp = BST;
if(!BST->Left && !BST->Right) // 没有孩子结点
BST = NULL;
else if(BST->Left && !BST->Right) // 只有左孩子结点
else if(!BST->Right) // 只有左孩子结点
BST = BST->Left;
else if(!BST->Left && BST->Right) // 只有右孩子结点
else if(!BST->Left) // 只有右孩子结点
BST = BST->Right;
free(tmp);
}
free(tmp);
}
return BST;
}
// 中序遍历
// 中序遍历
void InOrderTraversal(BinTree BT){
if(BT){
InOrderTraversal(BT->Left); // 进入左子树
cout<<BT->Data; // 打印根
InOrderTraversal(BT->Right); // 进入右子树
InOrderTraversal(BT->Left); // 进入左子树
cout<<BT->Data; // 打印根
InOrderTraversal(BT->Right); // 进入右子树
}
}
int main(){
@@ -125,14 +123,14 @@ int main(){
\ \
2 9
*/
cout<<"中序遍历的结果是:";
cout<<"中序遍历的结果是:";
InOrderTraversal(BST);
cout<<endl;
cout<<"查找最小值是:"<<FindMin(BST)->Data<<endl;
cout<<"查找最大值是:"<<FindMax(BST)->Data<<endl;
cout<<"查找值为3的结点左子树结点值为"<<Find(3,BST)->Left->Data<<endl;
cout<<"查找值为7的结点右子树结点值为"<<IterFind(7,BST)->Right->Data<<endl;
cout<<"删除值为5的结点"<<endl;
cout<<"查找最小值是:"<<FindMin(BST)->Data<<endl;
cout<<"查找最大值是:"<<FindMax(BST)->Data<<endl;
cout<<"查找值为3的结点左子树结点值为"<<Find(3,BST)->Left->Data<<endl;
cout<<"查找值为7的结点右子树结点值为"<<IterFind(7,BST)->Right->Data<<endl;
cout<<"删除值为5的结点"<<endl;
Delete(5,BST);
/*
6
@@ -143,7 +141,7 @@ int main(){
\ \
2 9
*/
cout<<"中序遍历的结果是:";
cout<<"中序遍历的结果是:";
InOrderTraversal(BST);
cout<<endl;
return 0;