B. DS二叉排序树之查找

题目描述

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

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

输入

第一行输入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;
}