ZJNU数据结构1017实验七 BST树的构建与应用 二叉排序树

Description

实现BST查找及建立算法，输入n个元素的关键字，建一颗基于这些关键字的BST树，重复的关键字不插入BST树中。

Input

第一行一个整数N,表示要输入的关键字的个数，第二行N个数，表示N个关键字。（100<=N<=10000）

Output

对于每组测试数据，输出两行，第一行为BST树中元素的个数
第二行为BST树的先序序列

Sample Input

6

12 23 32 4 8 12

Sample Output

5

12 4 8 23 32

代码区：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <string.h>
using namespace std;
typedef struct BSTNode//重命名结构体，方便使用
{
    int data;//存放数据
    struct BSTNode *lchild, *rchild;//左子树指针和右子树指针
}BSTNode,*BSTTree;//前者为结点，后者为树指针
int count=0;//初始化计数器
BSTTree root=NULL;//头结点初始化为NULL

void InsertBSTree(int m)//插入函数
{
    BSTTree p1,p2;
    if(root==NULL)//当头结点为空时开辟空间存放数据，并使左右子树指向NULL
    {
        root=(BSTTree)malloc(sizeof(BSTNode));//前面的括号将开辟的空间强制转化为树指针的格式
        root->data=m;
        root->lchild=root->rchild=NULL;
        count++;//计数器加一
    }
    else
    {
        p1=root;//运用另一个变量防止头结点发生变化（要不然就找不到了）
        while(m!=p1->data)//防止存放入相同的数据
        {
            if((m<p1->data)&&(p1->lchild!=NULL))
                //当该数据小于结点数据时并且左子树不为空，那么进入该结点的左子树进行比较
                //因为二叉排序树是左子树小于根节点，根节点小于右子树的
                p1=p1->lchild;
            else if((m>p1->data)&&(p1->rchild!=NULL))//该数据大于结点数据并右子树不为空
                p1=p1->rchild;
            else if((m<p1->data)&&(p1->lchild==NULL))//当小于结点数据且左子树为空时创建空间存放数据
            {
                p2=(BSTTree)malloc(sizeof(BSTNode));
                p2->data=m;
                p2->lchild=p2->rchild=NULL;
                p1->lchild=p2;
                count++;//计数器加一
                break;//该数据已存放成功，退出循环
            }
            else if((m>p1->data)&&(p1->rchild==NULL))//当大于节点数据且右子树为空时创建空间存放数据
            {
                p2=(BSTTree)malloc(sizeof(BSTNode));
                p2->data=m;
                p2->lchild=p2->rchild=NULL;
                p1->rchild=p2;
                count++;//计数器加一
                break;//该数据已存放成功，退出循环
            }
        }
    }
}
void CreateBSTTree()//初始化函数，输入数据
{
    int N;
    int a[10005];
    memset(a,0,sizeof(a));//初始化数组
    cin >> N;
    for(int i=0;i<N;i++)
    {
        cin >> a[i];
        InsertBSTree(a[i]);//插入
    }
}
void TravelBSTTree(BSTTree u)//遍历
{
    if(u!=NULL)//当不是空结点时
    {
        cout << u->data << " ";
        TravelBSTTree(u->lchild);//递归左子树
        TravelBSTTree(u->rchild);//递归右子树
        //先输出再依次递归左右子树为先序排列
    }
}
int main()
{
    CreateBSTTree();//调用函数
    cout << count << endl;//输出元素个数
    TravelBSTTree(root);//根据头结点来输出先序序列
    cout << endl;
    return 0;
}

新手上路，有错请指正