DS二叉排序树之查找

题目

问题 B: DS二叉排序树之查找
时间限制: 1 Sec  内存限制: 128 MB
提交: 358  解决: 302
[提交][状态][讨论版]
题目描述
给出一个数据序列，建立二叉排序树，并实现查找功能

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

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

代码块

#include <iostream>
using namespace std;

class BSTNode
{
private:
    int data;
    BSTNode *lchild, *rchild;
public:
    BSTNode(){lchild = NULL; rchild = NULL;}
    friend class BSTree;
};

class BSTree
{
private:
    BSTNode *root;
    void BST_insert(BSTNode *&p, int key);
    void InOrderTraverse(BSTNode *p);
    void InOrder();
    int BST_search(BSTNode *p, int key, int &count);
public:
    BSTree();
    void BST_create();
    void BST_search();
};

BSTree::BSTree()
{
    root = NULL;
}

void BSTree::InOrderTraverse(BSTNode *p)
{
    if(p)
    {
        InOrderTraverse(p->lchild);
        cout<<p->data<<' ';
        InOrderTraverse(p->rchild);
    }
}

void BSTree::InOrder()
{
    InOrderTraverse(root);
    cout<<endl;
}

void BSTree::BST_insert(BSTNode *&p, int key)
{
    if(!p)
    {
        p = new BSTNode;
        p->data = key;
    }
    else if(key<p->data)
        BST_insert(p->lchild, key);
    else
        BST_insert(p->rchild, key);
}

void BSTree::BST_create()
{
    int i, n, key;
    cin>>n;
    for(i=0; i<n; i++)
    {
        cin>>key;
        BST_insert(root, key);
    }
    InOrder();
}

int BSTree::BST_search(BSTNode *p, int key, int &count)
{
    if(!p)
        return 0;
    else if(key==p->data)
    {
        count++;
        return count;
    }
    else if(key<p->data)
    {
        count++;
        return BST_search(p->lchild, key, count);
    }
    else
    {
        count++;
        return BST_search(p->rchild, key, count);
    }
}

void BSTree::BST_search()
{
    int i, n, key, count;
    cin>>n;
    for(i=0; i<n; i++)
    {
        cin>>key;
        count = 0;
        if(BST_search(root, key, count))
        {
            count = 0;
            cout<<BST_search(root, key, count)<<endl;
        }
        else
            cout<<"-1"<<endl;
    }
}

int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        BSTree myBSTree;
        myBSTree.BST_create();
        myBSTree.BST_search();
    }
}