DS二叉排序树之删除

题目

问题 C: DS二叉排序树之删除
时间限制: 1 Sec  内存限制: 128 MB
提交: 497  解决: 288
[提交][状态][讨论版]
题目描述
给出一个数据序列，建立二叉排序树，并实现删除功能

对二叉排序树进行中序遍历，可以得到有序的数据序列

输入
第一行输入t，表示有t个数据序列

第二行输入n，表示首个序列包含n个数据

第三行输入n个数据，都是自然数且互不相同，数据之间用空格隔开

第四行输入m，表示要删除m个数据

从第五行起，输入m行，每行一个要删除的数据，都是自然数

以此类推输入下一个示例

输出
第一行输出有序的数据序列，对二叉排序树进行中序遍历可以得到

从第二行起，输出删除第m个数据后的有序序列，输出m行

以此类推输出下一个示例的结果

样例输入
1
6
22 33 55 66 11 44
3
66
22
77
样例输出
11 22 33 44 55 66 
11 22 33 44 55 
11 33 44 55 
11 33 44 55 
提示
当删除数据不在序列中，那么删除操作等于不执行，所以输出序列不变化

代码块

#include <iostream>
using namespace std;

class BSTNode
{
private:
    int data;
    BSTNode *lchild, *rchild;
public:
    BSTNode(){lchild = NULL; rchild = NULL;}
    friend class BSTree;
};

class BSTree
{
private:
    BSTNode *root;
    void BST_insert(BSTNode *&p, int key);
    void InOrderTraverse(BSTNode *p);
    void InOrder();
    void BST_delete(BSTNode *&p);
    void BST_delete(BSTNode *&p, int key);
public:
    BSTree();
    void BST_create();
    void BST_delete();
};

BSTree::BSTree()
{
    root = NULL;
}

void BSTree::InOrderTraverse(BSTNode *p)
{
    if(p)
    {
        InOrderTraverse(p->lchild);
        cout<<p->data<<' ';
        InOrderTraverse(p->rchild);
    }
}

void BSTree::InOrder()
{
    InOrderTraverse(root);
    cout<<endl;
}

void BSTree::BST_insert(BSTNode *&p, int key)
{
    if(!p)
    {
        p = new BSTNode;
        p->data = key;
    }
    else if(key<p->data)
        BST_insert(p->lchild, key);
    else
        BST_insert(p->rchild, key);
}

void BSTree::BST_create()
{
    int i, n, key;
    cin>>n;
    for(i=0; i<n; i++)
    {
        cin>>key;
        BST_insert(root, key);
    }
    InOrder();
}

void BSTree::BST_delete(BSTNode *&p)
{//找到要删除结点的位置后用该函数具体地将该节点删除
    BSTNode *q;
    if(!p->lchild && !p->rchild)
    {//叶子结点直接删除就好了
        delete p;
        p = NULL;
    }
    else if(!p->lchild && p->rchild)
    {//
        q = p;
        p = p->rchild;
        delete q;
    }
    else if(!p->rchild && p->lchild)
    {
        q = p;
        p = p->lchild;
        delete q;
    }
    else
    {
        BSTNode *parent = p;
        q = p->lchild;
        while(q->rchild)
        {
            parent = q;
            q = q->lchild;
        }
        p->data = q->data;
        if(parent==p)
            parent->lchild = q->lchild;
        else
            parent->rchild = q->lchild;
        delete q;
    }
}

void BSTree::BST_delete(BSTNode *&p, int key)
{//递归函数找到要删除结点的位置
    if(p)
    {
        if(key==p->data)
            BST_delete(p);
        else if(key<p->data)
            BST_delete(p->lchild, key);
        else
            BST_delete(p->rchild, key);
    }
}

void BSTree::BST_delete()
{//外包装函数
    int i, n, key;
    BSTNode *p;
    cin>>n;
    for(i=0; i<n; i++)
    {
        cin>>key;
        p = root;
        BST_delete(p, key);
        InOrder();
    }
}

int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        BSTree myBSTree;
        myBSTree.BST_create();
        myBSTree.BST_delete();
    }
}