浙大数据结构习题笔记：二叉搜索树

二叉搜索树

感觉FindTail和FindFor部分返回的内容不太好输出…

#include <iostream>
#include <malloc.h>
using namespace std;

typedef int ElemType;

typedef struct TreeNode *BinTree;
struct TreeNode{
    ElemType data;
    BinTree left;
    BinTree right;
};

BinTree FindTail(ElemType x,BinTree BST)
{
    if(!BST){
        return NULL;
    }
    if(x > BST->data){
        //尾递归
        return FindTail(x,BST->right);
    }else if(x < BST->data){
        return FindTail(x,BST->left);
    }else{
        return BST;
        //找到
    }
}

BinTree FindFor(ElemType x,BinTree BST)
{
    while(!BST){
        if(x > BST->data){
            BST = BST->right;
        }else if(x < BST->data){
            BST = BST->left;
        }else{
            return BST;
        }
    }
    return NULL;
}

//递归和循环，最大最小算法只需更换左右
BinTree FindMin(BinTree BST)
{
    if(!BST){
        return NULL;
    }else if(!BST->left){
        return BST;
    }else{
        return FindMin(BST->left);
    }
}

BinTree FindMax(BinTree BST)
{
    if(BST){
        while(BST->right){
            BST = BST->right;
        }
    }
    return BST;
}

BinTree Insert(ElemType x,BinTree BST)
{
    if(!BST){
        BST = (BinTree)malloc(sizeof(struct TreeNode));
        BST->data = x;
        BST->left = BST->right = NULL;
    }else{
        if(x < BST->data){
            BST->left = Insert(x,BST->left);
        }else if(x > BST->data){
            BST->right = Insert(x,BST->right);
        }
    }
    return BST;
}

BinTree Delete(ElemType x,BinTree BST)
{
    BinTree temp;
    if(!BST){
        printf("Not found");
    }else if(x < BST->data){
        BST->left = Delete(x,BST->left);
    }else if(x > BST->data){
        BST->right = Delete(x,BST->right);
    }else //节点已经找到
        if(BST->left && BST->right){    //两个子节点
            temp = FindMin(BST->right);
            //从右子树中找到最小值
            BST->data = temp->data;
            BST->right = Delete(BST->data,BST->right);
            //在右子树继续递归删除此值
        }else{  //只有一个或无子节点
            temp = BST;
            if(!BST->left){
                BST = BST->right;
            }else if(!BST->right){
                BST = BST->left;
            }
            free(temp);
        }
    return BST;
}

void InOrder(BinTree BT)
{
    if(BT){
        InOrder(BT->left);
        printf("%d ",BT->data);
        InOrder(BT->right);
    }
}

int main(){
	BinTree BST = NULL;
	BST = Insert(5,BST); 
	BST = Insert(7,BST); 
	BST = Insert(3,BST); 
	BST = Insert(1,BST); 
	BST = Insert(2,BST); 
	BST = Insert(4,BST); 
	BST = Insert(6,BST); 
	BST = Insert(8,BST); 
	BST = Insert(9,BST); 
	/*
			    5
			   /\
			  3  7
             /\	 /\
            1 4 6  8
			\      \
			 2      9
	*/
	printf("InOrder"); 
    InOrder(BST);
    printf("\n");
	printf("Min Value:%d\n",FindMin(BST)->data);
	printf("Max Value:%d\n",FindMax(BST)->data); 
	//printf("Find 3 Value:%d\n",FindFor(3,BST)->data);
	//printf("Find 7 Value:%d\n",FindTail(7,BST)->data);
	printf("Delete 5\n");
	Delete(5,BST);
	/*
			    6
			   /\
			  3  7
             /\	  \
            1 4    8
			\      \
			 2      9
	*/
	printf("InOrder:"); 
	InOrder(BST);
	printf("\n");
	return 0;
}