题目描述
给出一个数据序列,建立二叉排序树,并实现查找功能
对二叉排序树进行中序遍历,可以得到有序的数据序列
输入
第一行输入t,表示有t个数据序列
第二行输入n,表示首个序列包含n个数据
第三行输入n个数据,都是自然数且互不相同,数据之间用空格隔开
第四行输入m,表示要查找m个数据
从第五行起,输入m行,每行一个要查找的数据,都是自然数
以此类推输入下一个示例
输出
第一行输出有序的数据序列,对二叉排序树进行中序遍历可以得到
从第二行起,输出查找结果,如果查找成功输出查找次数,如果查找失败输出-1
以此类推输出下一个示例的结果
输入样例
1
6
22 33 55 66 11 44
7
11
22
33
44
55
66
77
输出样例
11 22 33 44 55 66
2
1
2
4
3
4
-1
思路分析
建立二叉排序树的方法就是,一个一个地插入树中。插入数据的时候,从根结点开始比较,若数据大于结点的数据,就与右孩子比。若数据小于该结点的数据,就与左孩子比较。如果是null,就可以插入了。这里还是要用到递归。
查找功能也是有个递归函数,二叉树的题目核心都是递归的思想。每往下查找一次,查找次数就+1,所以递归函数返回int值。如果找到最下面发现没有这个数据,找不到,就返回一个恨负的负数值。最后返回的数值为负就说明没找到,为正即代表找到了。
AC代码
#include<iostream>
using namespace std;
class node{
public:
int data;
node *lchild,*rchild;
node():data(0){
lchild=NULL;
rchild=NULL;
}
friend class tree;
};
class tree{
private:
node *root;
public:
tree(){
root=new node();
}
node * getroot(){
return root;
}
node* insert(int x,node *r){
if(r->data==0){
r->data=x;
}
else if(x>r->data&&r->rchild!=NULL){
insert(x,r->rchild);
}
else if(x<r->data&&r->lchild!=NULL){
insert(x,r->lchild);
}
else if(x>r->data){
node *p=new node();
p->data=x;
r->rchild=p;
}
else{
node *p=new node();
p->data=x;
r->lchild=p;
}
return r;
}
void create(int n){
int x;
while(n--){
cin>>x;
insert(x,root);
}
}
void Search_func(int m){
int x;
while(m--){
cin>>x;
int num=search(x,root);
if(num<0) cout<<"-1"<<endl;
else cout<<num<<endl;
}
}
int search(int x,node *r){
if(r->data>x&&r->lchild==NULL) return -9999;
if(r->data<x&&r->rchild==NULL) return -9999;
if(r->data==x) return 1;
if(r->data>x) return 1+search(x,r->lchild);
if(r->data<x) return 1+search(x,r->rchild);
}
void zhongxu(node* r){
if(r->lchild!=NULL) zhongxu(r->lchild);
cout<<r->data<<" ";
if(r->rchild!=NULL) zhongxu(r->rchild);
}
};
int main(){
int t,n,m;
cin>>t;
tree Mytree;
while(t--){
cin>>n;
Mytree.create(n);
Mytree.zhongxu(Mytree.getroot());
cout<<endl;
cin>>m;
Mytree.Search_func(m);
}
return 0;
}