hdu 5444 继续二叉查找树

/*************************************************************************  这是一个二叉查找树，实现了以下操作：插入结点、构造二叉树、删除结点、查找、  查找最大值、查找最小值、查找指定结点的前驱和后继。上述所有操作时间复杂度  均为o(h),其中h是树的高度  注释很详细，具体内容就看代码吧*************************************************************************/#include<string>#include<iostream>#include<stdio.h>#include<stdlib.h>using namespace std;const int maxn = 1e5 + 10;//二叉查找树结点描述//typedef int KeyType;typedef struct Node{    int key;          //关键字    struct Node * left;   //左孩子指针    struct Node * right;  //右孩子指针    struct Node * parent; //指向父节点指针}Node,*PNode;//往二叉查找树中插入结点//插入的话，可能要改变根结点的地址，所以传的是二级指针void inseart(PNode * root,int key){    //初始化插入结点    PNode p=(PNode)malloc(sizeof(Node));    p->key=key;    p->left=p->right=p->parent=NULL;    //空树时，直接作为根结点    if((*root)==NULL){        *root=p;        return;    }    //插入到当前结点（*root）的左孩子    if((*root)->left == NULL && (*root)->key < key){        p->parent=(*root);        (*root)->left=p;        return;    }    //插入到当前结点（*root）的右孩子    if((*root)->right == NULL && (*root)->key > key){        p->parent=(*root);        (*root)->right=p;        return;    }    if((*root)->key < key)        inseart(&(*root)->left,key);    else if((*root)->key > key)        inseart(&(*root)->right,key);    else        return;}int a[maxn];string ans ;void search(PNode root,int key)  {      if(root == NULL)          return ;      if(key < root->key) //查找右子树      { ans = ans + "E"; search(root->right,key);}      else if(key > root->key) //查找左子树      { ans = ans + "W"; search(root->left,key); }     else          return ;  }int main(){    int t ; cin >> t; int q, n ;    while(t--){        PNode root=NULL;        scanf("%d", &n);        for(int i = 1; i <= n ; i ++){            scanf("%d", &a[i]);        }        for(int i = 1 ; i <= n ; i ++){            inseart(&root, a[i]);        }        scanf("%d", &q); int num;        while( q-- ){            scanf("%d", &num);            ans = "";            search(root , num);            cout << ans <<endl;        }    }}

/*************************************************************************  这是一个二叉查找树，实现了以下操作：插入结点、构造二叉树、删除结点、查找、  查找最大值、查找最小值、查找指定结点的前驱和后继。上述所有操作时间复杂度  均为o(h),其中h是树的高度  注释很详细，具体内容就看代码吧*************************************************************************/#include<stdio.h>#include<stdlib.h>//二叉查找树结点描述typedef int KeyType;typedef struct Node{    KeyType key;          //关键字    struct Node * left;   //左孩子指针    struct Node * right;  //右孩子指针    struct Node * parent; //指向父节点指针}Node,*PNode;//往二叉查找树中插入结点//插入的话，可能要改变根结点的地址，所以传的是二级指针void inseart(PNode * root,KeyType key){    //初始化插入结点    PNode p=(PNode)malloc(sizeof(Node));    p->key=key;    p->left=p->right=p->parent=NULL;    //空树时，直接作为根结点    if((*root)==NULL){        *root=p;        return;    }    //插入到当前结点（*root）的左孩子    if((*root)->left == NULL && (*root)->key > key){        p->parent=(*root);        (*root)->left=p;        return;    }    //插入到当前结点（*root）的右孩子    if((*root)->right == NULL && (*root)->key < key){        p->parent=(*root);        (*root)->right=p;        return;    }    if((*root)->key > key)        inseart(&(*root)->left,key);    else if((*root)->key < key)        inseart(&(*root)->right,key);    else        return;}//查找元素,找到返回关键字的结点指针，没找到返回NULLPNode search(PNode root,KeyType key){    if(root == NULL)        return NULL;    if(key > root->key) //查找右子树        return search(root->right,key);    else if(key < root->key) //查找左子树        return search(root->left,key);    else        return root;}//查找最小关键字,空树时返回NULLPNode searchMin(PNode root){    if(root == NULL)        return NULL;    if(root->left == NULL)        return root;    else  //一直往左孩子找，直到没有左孩子的结点        return searchMin(root->left);}//查找最大关键字,空树时返回NULLPNode searchMax(PNode root){    if(root == NULL)        return NULL;    if(root->right == NULL)        return root;    else  //一直往右孩子找，直到没有右孩子的结点        return searchMax(root->right);}//查找某个结点的前驱PNode searchPredecessor(PNode p){    //空树    if(p==NULL)        return p;    //有左子树、左子树中最大的那个    if(p->left)        return searchMax(p->left);    //无左子树,查找某个结点的右子树遍历完了    else{        if(p->parent == NULL)            return NULL;        //向上寻找前驱        while(p){            if(p->parent->right == p)                break;            p=p->parent;        }        return p->parent;    }}//查找某个结点的后继PNode searchSuccessor(PNode p){    //空树    if(p==NULL)        return p;    //有右子树、右子树中最小的那个    if(p->right)        return searchMin(p->right);    //无右子树,查找某个结点的左子树遍历完了    else{        if(p->parent == NULL)            return NULL;        //向上寻找后继        while(p){            if(p->parent->left == p)                break;            p=p->parent;        }        return p->parent;    }}//根据关键字删除某个结点,删除成功返回1,否则返回0//如果把根结点删掉，那么要改变根结点的地址，所以传二级指针int deleteNode(PNode* root,KeyType key){    PNode q;    //查找到要删除的结点    PNode p=search(*root,key);    KeyType temp;    //暂存后继结点的值    //没查到此关键字    if(!p)        return 0;    //1.被删结点是叶子结点，直接删除    if(p->left == NULL && p->right == NULL){        //只有一个元素，删完之后变成一颗空树        if(p->parent == NULL){            free(p);            (*root)=NULL;        }else{            //删除的结点是父节点的左孩子            if(p->parent->left == p)                p->parent->left=NULL;            else  //删除的结点是父节点的右孩子                p->parent->right=NULL;            free(p);        }    }    //2.被删结点只有左子树    else if(p->left && !(p->right)){        p->left->parent=p->parent;        //如果删除是父结点，要改变父节点指针        if(p->parent == NULL)            *root=p->left;        //删除的结点是父节点的左孩子        else if(p->parent->left == p)            p->parent->left=p->left;        else //删除的结点是父节点的右孩子            p->parent->right=p->left;        free(p);    }    //3.被删结点只有右孩子    else if(p->right && !(p->left)){        p->right->parent=p->parent;        //如果删除是父结点，要改变父节点指针        if(p->parent == NULL)            *root=p->right;        //删除的结点是父节点的左孩子        else if(p->parent->left == p)            p->parent->left=p->right;        else //删除的结点是父节点的右孩子            p->parent->right=p->right;        free(p);    }    //4.被删除的结点既有左孩子，又有右孩子    //该结点的后继结点肯定无左子树(参考上面查找后继结点函数)    //删掉后继结点,后继结点的值代替该结点    else{        //找到要删除结点的后继        q=searchSuccessor(p);        temp=q->key;        //删除后继结点        deleteNode(root,q->key);        p->key=temp;    }    return 1;}//创建一棵二叉查找树void create(PNode* root,KeyType *keyArray,int length){    int i;    //逐个结点插入二叉树中    for(i=0;i<length;i++)        inseart(root,keyArray[i]);}int main(void){    int i;    PNode root=NULL;    KeyType nodeArray[11]={15,6,18,3,7,17,20,2,4,13,9};    create(&root,nodeArray,11);    for(i=0;i<2;i++)        deleteNode(&root,nodeArray[i]);    printf("%d\n",searchPredecessor(root)->key);    printf("%d\n",searchSuccessor(root)->key);    printf("%d\n",searchMin(root)->key);    printf("%d\n",searchMax(root)->key);    printf("%d\n",search(root,13)->key);    return 0;}