是否同一棵二叉搜索树 (25 分)
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{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
我的思路是这样的,先构建一颗标准的二叉搜索树(下面AC代码中会给出),并用数组储存他的先序遍历,而后按要求建立L颗需要判断的树,每次都与标准的二叉树的先序遍历进行比较,如果不相同,则 不同,如果相同,则同。
注意:思路简单,但部分代码书写时会出现很多细节问题,比如在进行先序遍历的比较时递归函数应当如何书写。。。。。。此题推荐自己书写后,再看AC代码:
#include <bits/stdc++.h>
using namespace std;
int N,L,pos,samejudgement[100];
typedef struct node
{
node(int a,node* b,node* c)
{
data=a;
l=b;
r=c;
}
int data;
struct node *l;
struct node *r;
}*BST;
void PreOrderTraverse(BST T)
{
if(T)
{
samejudgement[pos++]=T->data;
PreOrderTraverse(T->l);
PreOrderTraverse(T->r);
}
}
bool PreOrderTraverse_test(BST T)
{
if(T)
{
// cout<<"gg1111 "<<samejudgement[pos]<<" "<<pos<<endl;
if(samejudgement[pos++]!=T->data) return false;
// cout<<"gg2222 "<<pos<<endl;
if(!PreOrderTraverse_test(T->l)) return false;
if(!PreOrderTraverse_test(T->r)) return false;
}
return true;
}
void Inset_BST(BST &T,int data)
{
if(T==NULL)
{
T=new node(data,NULL,NULL);
}
else if(T->data > data)
{
Inset_BST(T->l,data);
}
else Inset_BST(T->r,data);
}
int main()
{
while(scanf("%d",&N)!=EOF)
{
if(N) scanf("%d",&L);
else break;
pos=0;
memset(samejudgement,0,sizeof(samejudgement));
BST tree=NULL;
for(int i=0; i<N; i++)
{
int t; scanf("%d",&t);
Inset_BST(tree,t);
}
PreOrderTraverse(tree);
//for(int i=0;i<pos;i++) cout<<"输出中间值 "<<samejudgement[i]<<endl;
bool flag=true;
for(int i=0;i<L;i++)
{
flag=true;
BST test_tree=NULL;
pos=0;
for(int j=0;j<N;j++)
{
int t; scanf("%d",&t);
Inset_BST(test_tree,t);
}
//PreOrderTraverse_test(test_tree);
if(!PreOrderTraverse_test(test_tree)) flag=false;
if(flag) printf("Yes\n");
else printf("No\n");
}
//if(flag) printf("Yes\n");
//else printf("No\n");
}
return 0;
}