数据结构实验之查找一：二叉排序树

数据结构实验之查找一：二叉排序树

Time Limit: 400MS Memory Limit: 65536KB

Submit Statistic

Problem Description

对应给定的一个序列可以唯一确定一棵二叉排序树。然而，一棵给定的二叉排序树却可以由多种不同的序列得到。例如分别按照序列{3,1,4}和{3,4,1}插入初始为空的二叉排序树，都得到一样的结果。你的任务书对于输入的各种序列，判断它们是否能生成一样的二叉排序树。

Input

输入包含若干组测试数据。每组数据的第1行给出两个正整数N (n < = 10)和L，分别是输入序列的元素个数和需要比较的序列个数。第2行给出N个以空格分隔的正整数，作为初始插入序列生成一颗二叉排序树。随后L行，每行给出N个元素，属于L个需要检查的序列。
简单起见，我们保证每个插入序列都是1到N的一个排列。当读到N为0时，标志输入结束，这组数据不要处理。

Output

对每一组需要检查的序列，如果其生成的二叉排序树跟初始序列生成的二叉排序树一样，则输出"Yes"，否则输出"No"。

Example Input

4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0

Example Output

Yes
No
No

所谓二叉排序树就是对于每个结点左孩子的值必定小于结点的值而结点的值必定小于右孩子的值。

所以建树的时候，就是比结点小就去左（子树）孩子插，比结点大就去右（子树）孩子插。

这个题我们只要建出树来，然后输出一次先序或者别的也行。只要遍历的结果一样，就表示这两棵树是一样的。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n,m;
int cnt;
int tree1[15],tree2[15];
struct node
{
int num;
struct node*left,*right;
};
struct node* creat()
{
struct node*p=new node;
p->left=p->right=NULL;
return p;
};
void add(struct node*&root,int x)
{
if(!root)
{
root=creat();
root->num=x;
return;
}
if(x<root->num)add(root->left,x);
else add(root->right,x);
};
void get_pre(struct node *&root,int *a)
{
if(!root)return;
a[cnt++]=root->num;
if(root->left)get_pre(root->left,a);
if(root->right)get_pre(root->right,a);
}
bool check()
{
for(int i=0;i<n;++i)
{
if(tree1[i]!=tree2[i])return 0;
}
return 1;
}
int main()
{
while(~scanf("%d",&n)&&n)
{
scanf("%d",&m);
int i;
struct node *root=NULL;
int x;
for(i=0; i<n; ++i)
{
scanf("%d",&x);
add(root,x);
}
cnt=0;
get_pre(root,tree1);
while(m--)
{
struct node*root2=NULL;
for(i=0; i<n; ++i)
{
scanf("%d",&x);
add(root2,x);
}
cnt=0;
get_pre(root2,tree2);
if(check())puts("Yes");
else puts("No");
}
}
return 0;
}