SWUST OJ 1011: 二叉排序树的实现和查找

1011: 二叉排序树的实现和查找

题目链接-1011: 二叉排序树的实现和查找

解题思路
二叉排序树：一棵空树，或者是具有下列性质的二叉树：
   （1）若左子树不空，则左子树上所有结点的值均小于它的根结点的值；
   （2）若右子树不空，则右子树上所有结点的值均大于它的根结点的值；
   （3）左、右子树也分别为二叉排序树；
   （4）没有键值相等的结点

• 建树时我们采用插入建树的方法，若当前结点为空，则建立根节点，若插入的值小于当前父亲结点的值，则作为左节点插入，否则就作为右节点插入
• 查找时若子树为空，查找不成功，若小于根结点的值，递归查左子树，若大于根结点的值，递归查右子树
• 可以用$cnt$记录查找次数，即调用函数的次数
• 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
struct node{
	int data;
	node *l,*r;
}*p;
void build(node *&p,int x){
	if(p==NULL){
		p=new node;
		p->data=x;
		p->l=NULL;
		p->r=NULL;
	}
	else if(x<p->data) build(p->l,x);
	else if(x>p->data) build(p->r,x);
}
int cnt;
node *search(node *&p,int x){
	cnt++;
	if(p==NULL||p->data==x) return p;
	else if(x<p->data) search(p->l,x);
	else if(x>p->data) search(p->r,x);
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);

	int n,key;
	cin>>n;
	for(int i=0;i<n;i++){
		int tmp;
		cin>>tmp;
		build(p,tmp);
	}
	cin>>key;
	if(search(p,key)==NULL) cout<<-1;
	else cout<<cnt;
	return 0;
}