swust oj 1011

二叉排序树的实现和查找

1000(ms)

10000(kb)

2782 / 6301

按照给定的关键字集合，建立二叉排序树。在建立的二叉排序树上查找指定的关键字，查找成功，输出找到该关键字比较的次数；查找不成功，输出-1.

输入

关键字个数n； 
关键字集合； 
要查找的关键字；

输出

查找成功输出比较的次数，否则输出-1。

样例输入

12
25 18 46 2 53 39 32 4 74 67 60 11
74

样例输出

4

 

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cstdio>
typedef int Datetype;
using namespace std;
int x;

typedef struct link{
    Datetype date;
    struct link *lchild;
    struct link *rchild;
}tree;

void creat(tree *&L, int arr[] ,int l,int r)
{
    if(l>r)
    {
        L=NULL;
        return ;
    }
    int s=(l+r)>>1;
    L=new tree;
    L->date=arr[s];
    creat(L->lchild,arr,l,s-1);
    creat(L->rchild,arr,s+1,r);
}

void display(tree *&L)
{
    if(L!=NULL)
    {
        cout<<L->date<<" ";
        display(L->lchild);
        display(L->rchild);
    }
}

void delet(tree *&L)
{
    if(L!=NULL)
    {
        delet(L->lchild);
        delet(L->rchild);
        delete(L);
    }
}

void find(tree *&L, int a)
{
    if(L!=NULL)
    {
        x++;
        if(a>L->date)
            find(L->rchild,a);
        else if(a<L->date)
            find(L->lchild,a);
        else
            return ;
    }
    if(L==NULL)
        x=0;
}

int main()
{
    int arr[100];
    int n,a;
    tree *L;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>arr[i];
    }
    sort(arr,arr+n);
    cin>>a;
    creat(L,arr,0,n-1);
//    display(L);
    find(L,a);
    if(x)
        cout<<x;
    else
        cout<<"-1";
    delet(L);
    return 0;
}