04-树4 是否同一棵二叉搜索树(25 分)

题目来源:中国大学MOOC-陈越、何钦铭-数据结构-2018春
作者: 陈越
单位: 浙江大学

问题描述:

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。最后L行,每行给出N个插入的元素,属于L个需要检查的序列。
简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。
输出格式:
对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。
输入样例:
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
输出样例:
Yes
No
No

解答:建树,判断。在树节点的结构中可以添加一个flag变量表示该节点是否被访问过,在判断是否是同一棵二叉树遍历时,如果出现节点flag为false且节点的值与要判断的值不等,则此序列代表的二叉搜索树肯定与原树不一致,返回,注意每次执行完一组要reset每个节点的flag值。
要注意递归建树过程,返回一个节点要用父节点的左子树或者右子树接收它,才能保证树完整建立。

#include <iostream>

using namespace std;
struct node
{
    int value;
    node* rightChild;
    node* leftChild;
    int isVisit;
    node()
    {
        rightChild=NULL;
        leftChild=NULL;
        isVisit=false;
    }
};
node* buildTree(int value,node* head)
{
    if(head==NULL)
    {
        node* temp=new node();
        temp->value=value;
        return temp;
    }
    if(value>head->value)
    {
        head->rightChild=buildTree(value,head->rightChild);
    }
    else
    {
        head->leftChild=buildTree(value,head->leftChild);
    }
    return head;
}
int N,L;
void resetFlag(node* head)
{
    if(head->leftChild!=NULL)
        resetFlag(head->leftChild);
    if(head->rightChild!=NULL)
        resetFlag(head->rightChild);
    head->isVisit=false;
}
bool judge(int value,node* head)
{
    if(head->isVisit==false)
    {
        if(head->value==value)
        {
            head->isVisit=true;
            return true;
        }
        else
            return false;
    }
    else
    {
        if(value>head->value)
            judge(value,head->rightChild);
        else
            judge(value,head->leftChild);
    }
}
int main()
{
    while(1)
    {
        cin>>N;
        if(N==0)
            break;
        cin>>L;
        node* tree=NULL;
        for(int i=1;i<=N;i++)
        {
            int value;
            cin>>value;
            tree=buildTree(value,tree);
        }
        for(int i=1;i<=L;i++)
        {
            bool flag=true;
            resetFlag(tree);
            for(int j=1;j<=N;j++)
            {
                int value;
                cin>>value;
                if(!judge(value,tree))
                {
                    flag=false;
                }
            }
            if(!flag)
                cout<<"No"<<endl;
            else
                cout<<"Yes"<<endl;
        }
    }
    return 0;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/peterchen96/article/details/79975458
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭