题目描述
给出一个数据序列,建立二叉排序树,并实现删除功能
对二叉排序树进行中序遍历,可以得到有序的数据序列
输入
第一行输入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;
}