DS二叉排序树之删除

题目描述

给出一个数据序列，建立二叉排序树，并实现删除功能

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

输入

第一行输入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 "bits/stdc++.h"
using namespace std;
int t,n;
struct NODE{
    int data;
    NODE *l,*r;
    NODE(){
        l=NULL;
        r=NULL;
    }
};

void display(NODE *head){
    if(head->l) display(head->l);
    cout<<head->data<<" ";
    if(head->r) display(head->r);
}

void build(NODE *&head,int x){
    if(!head){
        head=new NODE;
        head->data=x;
        return ;
    }
    if(x>head->data){
        build(head->r,x);
    }
    else{
        build(head->l,x);
    }
}

void del(NODE *&head,int x){
    if(head==NULL) return;
    if(head->data==x){
        if(!head->l && !head->r) head=NULL;
        else if(!head->l){
            head->data=head->r->data;
            head->l=head->r->l;
            head->r=head->r->r;
        }
        else if(!head->r){
            head->data=head->l->data;
            head->r=head->l->r;
            head->l=head->l->l;
        }
        else{
            NODE *node=head->l;
            while(node->r){
                node=node->r;
            }
            head->data=node->data;
            del(head->l,node->data);
        }
    }
    else if(x<head->data){
        del(head->l,x);
    }
    else del(head->r,x);
}

int main(){
//    freopen("123.in","r",stdin);
    cin>>t;
    while(t--){
        cin>>n;
        NODE *head=NULL;
        for(int i=1;i<=n;i++){
            int x;
            cin>>x;
            build(head,x);
        }
        display(head);
        cout<<endl;

        int m;
        cin>>m;
        for(int i=1;i<=m;i++){
            int x;
            cin>>x;
            del(head,x);
            display(head);
            cout<<endl;
        }
    }
    return 0;
}