7-11 是否同一棵二叉搜索树 (25 分)
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数 ()和,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出个以空格分隔的正整数,作为初始插入序列。最后行,每行给出个插入的元素,属于个需要检查的序列。
简单起见,我们保证每个插入序列都是1到的一个排列。当读到为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
鸣谢青岛大学周强老师补充测试数据!
#include<bits/stdc++.h>
using namespace std;
int n,l,num;
void Insert(int *a,int root,int num)
{
if(a[root]==-1)
{
a[root] = num;return;
}
else if(num<a[root])Insert(a,root<<1,num);
else Insert(a,root<<1|1,num);
}
int cmp(int *a,int *b)
{
for(int i=1; i<=100; i++)
{
if(a[i]!=b[i])return 0;
}
return 1;
}
int a[2090],b[2090];
int main()
{
while(cin>>n&&n)
{
memset(a,-1,sizeof(a));
cin>>l;
for(int i=1; i<=n; i++)
{
cin>>num;
Insert(a,1,num);
}
while(l--)
{
memset(b,-1,sizeof(b));
for(int i=1; i<=n; i++)
{
cin>>num;
Insert(b,1,num);
}
if(cmp(a,b))printf("Yes\n");
else printf("No\n");
}
}
return 0;
}CODE2
#include<bits/stdc++.h>
using namespace std;
int a[1231234],b[1231234];
void Insert(int *a,int root,int num)
{
if(a[root]==0)
{
a[root] = num;
return ;
}
else if(num<a[root])Insert(a,root<<1,num);
else Insert(a,root<<1|1,num);
}
int judge(int root1,int root2)
{
if(!a[root1]&&!b[root2])return 1;
if(a[root1]!=b[root2])return 0;
else return judge(root1<<1,root2<<1)&&judge(root1<<1|1,root2<<1|1);
}
int main()
{
int n,m,g;
while(cin>>n)
{
if(n==0)break;
cin>>m;
memset(a,0,sizeof a);
for(int i=1; i<=n; i++)
{
scanf("%d",&g);
Insert(a,1,g);
}
while(m--)
{
memset(b,0,sizeof b);
for(int i=1; i<=n; i++)
{
scanf("%d",&g);
Insert(b,1,g);
}
int f = judge(1,1);
if(f)printf("Yes\n");
else printf("No\n");
}
}
}