数据结构实验之查找一:二叉排序树
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
Hint
Author
xam
#include <iostream>
using namespace std;
const int MAX = 15;
typedef struct node
{
int d;
struct node *r, *l;
} tree, *ptree;
ptree creat(ptree a, int x)
{
if(a == NULL)
{
a = new tree();
a->d = x;
a->r = a->l = NULL;
}
else
{
if(x < a->d)
{
a->l = creat(a->l, x);
}
else
{
a->r = creat(a->r, x);
}
}
return a;
}
int judge(ptree a, ptree b)
{
if(a == NULL && b == NULL)
return 1;
else if(a != NULL && b != NULL)
{
if(a->d != b->d)
return 0;
else if(judge(a->l, b->l) && judge(a->r, b->r))
return 1;
else return 0;
}
else return 0;
}
int main()
{
int n, m, i;
int ra[MAX], rb[MAX];
while(cin >> n >> m && n != 0)
{
for(i = 0; i < n; i++)
cin >> ra[i];
ptree a = new tree();
a->d = ra[0];
a->l = a->r = NULL;
for(i = 1; i < n; i++)
a = creat(a, ra[i]);
while(m--)
{
for(i = 0; i < n; i++)
cin >> rb[i];
ptree b = new tree();
b->d = rb[0];
b->l = b->r = NULL;
for(i = 1; i < n; i++)
b = creat(b, rb[i]);
int k = judge(a, b);
if(k == 1)
cout << "Yes" << endl;
else cout << "No" << endl;
}
}
return 0;
}