题目来源:中国大学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;
}