PTA 7-5 二叉搜索树的结构 (30 分)

PTA 7-5 二叉搜索树的结构 (30 分)

二叉搜索树或者是一棵空树，或者是具有下列性质的二叉树： 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；它的左、右子树也分别为二叉搜索树。（摘自百度百科）
给定一系列互不相等的整数，将它们顺次插入一棵初始为空的二叉搜索树，然后对结果树的结构进行描述。你需要能判断给定的描述是否正确。例如将{ 2 4 1 3 0 }插入后，得到一棵二叉搜索树，则陈述句如“2是树的根”、“1和4是兄弟结点”、“3和0在同一层上”（指自顶向下的深度相同）、“2是4的双亲结点”、“3是4的左孩子”都是正确的；而“4是2的左孩子”、“1和3是兄弟结点”都是不正确的。

在二叉树的结构中增加一个用于指向父亲节点的指针域，并在构建树的时候就记录树的层次

#include <iostream>
using namespace std;
template<typename T>
struct TreeNode
{
    T data;
    TreeNode<T> *lchild=NULL;
    TreeNode<T> *rchild=NULL;
    TreeNode<T> *parent=NULL;
    int level;
    TreeNode(T n)
    {
        data=n;
        lchild=NULL;
        rchild=NULL;
        parent=NULL;
    }
};
template<typename T>
void AddNode(TreeNode<T> *&subroot,const T&elem)
{
    if(subroot==NULL)
    {
        subroot=new TreeNode<int>(elem);
        subroot->level=0;
    }
    else
    {
        TreeNode<T> *pre=subroot;
        int c=1;
        T t=pre->data;
        while(true)
        {
            t=pre->data;
            if(elem<t)
            {
                if(pre->lchild!=NULL)
                {
                    c++;
                    pre=pre->lchild;
                }
                else
                {
                    pre->lchild=new TreeNode<T>(elem);
                    pre->lchild->parent=pre;
                    pre->lchild->level=c+1;
                    break;
                }
            }
            else
            {
                if(pre->rchild!=NULL)
                {
                    c++;
                    pre=pre->rchild;
                }
                else
                {
                    pre->rchild=new TreeNode<T>(elem);
                    pre->rchild->parent=pre;
                    pre->rchild->level=c+1;
                    break;
                }
            }
        }
    }
}
template<typename T>
TreeNode<T> *Search(TreeNode<T> *&subroot,T elem)
{
    if(subroot==NULL)
    {
        return NULL;
    }
    else if(subroot->data==elem)
    {
        return subroot;
    }
    else
    {
        TreeNode<T> *pre=subroot;
        while(pre!=NULL)
        {
            if(pre->data<elem)
            {
                pre=pre->rchild;
            }
            else if(pre->data>elem)
            {
                pre=pre->lchild;
            }
            else
            {
                break;
            }
        }
        return pre;
    }
}
int main()
{
    TreeNode<int> *tree=NULL,*p,*q;
    int n,num;
    string ss;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        cin>>num;
        AddNode(tree,num);
    }
    cin>>n;
    getline(cin,ss);
    for(int i=0; i<n; i++)
    {
        int t;
        cin>>t;
        cin>>ss;
        if(ss=="is")
        {
            cin>>ss;
            cin>>ss;
            if(ss=="root")
            {
                if(tree!=NULL&&tree->data==t)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
            else if(ss=="parent")
            {
                int t2;
                cin>>ss;
                cin>>t2;
                p=Search(tree,t2);
                if(p!=NULL&&p->parent!=NULL&&p->parent->data==t)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
            else if(ss=="left")
            {
                int t2;
                cin>>ss;
                cin>>ss;
                cin>>t2;
                q=Search(tree,t2);
                if(q!=NULL&&q->lchild!=NULL&&q->lchild->data==t)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
            else if(ss=="right")
            {
                int t2;
                cin>>ss;
                cin>>ss;
                cin>>t2;
                q=Search(tree,t2);
                if(q!=NULL&&q->rchild!=NULL&&q->rchild->data==t)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
        }
        else if(ss=="and")
        {
            int t2;
            cin>>t2;
            getline(cin,ss);
            if(ss.find("siblings")!=string::npos)
            {
                p=Search(tree,t);
                q=Search(tree,t2);
                if(p!=NULL&&q!=NULL&&p->parent==q->parent)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
            else if(ss.find("level")!=string::npos)
            {
                p=Search(tree,t);
                q=Search(tree,t2);
                if(p!=NULL&&q!=NULL&&p->level==q->level)
                {
                    cout<<"Yes"<<endl;
                }
                else
                {
                    cout<<"No"<<endl;
                }
                continue;
            }
        }
    }
    return 0;
}