机试算法模板--二叉搜索树

二叉搜索树

#include <iostream>
using namespace std;
struct node
{
    int data;
    node *lchild;
    node *rchild;
};

//新建节点
node *create(int v)
{
    node *Node = new node; //申请一个node型变量的地址空间
    Node->data = v;
    Node->lchild = NULL; //必须加这句话
    Node->rchild = NULL;
    return Node; //返回新建立的节点的地址
}

//查找
void search(node *root, int x)
{
    if (root == NULL)
    {
        return; //空树，死胡同(递归边界)
    }
    if (root->data == x)
    {
        cout << root->data << endl;
    }
    else if (root->data > x)
    {
        search(root->lchild, x);
    }
    else
    {
        search(root->rchild, x);
    }
}

//插入
void insert(node *&root, int x)
{
    if (root == NULL) //递归边界
    {
        root = create(x);
        return;
    }
    if (root->data == x)
    {
        return; //查找成功
    }
    else if (root->data > x)
    {
        insert(root->lchild, x);
    }
    else
    {
        insert(root->rchild, x);
    }
}

//二叉搜索树的创建
node *build(int data[], int n)
{
    node *root = NULL; //新建空根节点root
    for (int i = 0; i < n; i++)
    {
        insert(root, data[i]);
    }
    return root;
}

//寻找以root为根节点的树中的最大权值节点
node *findMax(node *root)
{
    while (root->lchild)
    {
        root = root->lchild;
    }
    return root;
}

//寻找以root为根节点的树中的最小全权值节点
node *findMin(node *root)
{
    while (root->rchild)
    {
        root = root->rchild;
    }
    return root;
}

//删除以root为根节点的树中权值为x的结点
void deleteNode(node *root, int x)
{
    if(root == NULL){
        return;
    }
    if(root->data == x){
        if(root->lchild == NULL && root->rchild == NULL){
            root = NULL;
        }else if(root->lchild){
            node* temp = findMax(root->lchild);
            root->data = temp->data;
            deleteNode(root->lchild,temp->data);
        }else{
            node *temp = findMin(root->rchild);
            root->data = temp->data;
            deleteNode(root->rchild, temp->data);
        }
    }else if(root->data > x){
        deleteNode(root->lchild,x);
    }else{
        deleteNode(root->rchild,x);
    }
}