Time Limit: 400 ms Memory Limit: 65536 KiB
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"。
Sample Input
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
Sample Output
Yes
No
No
思路:
定义两棵树,把两棵树按题中所给数据建好,而后在四中输出顺序中选出一种,按顺序存入一个数组中,然后比较两个数组是否相投,若相同则树相同。
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
typedef struct node
{
int data;
struct node *lt, *rt;
}BT;
int a[60], b[60], flag, flag1;
void creat(BT *&root, int x)
{
if(!root)
{
root = (BT *)malloc(sizeof(BT));
root->lt = root->rt = NULL;
root->data = x;
}
else
{
if(x > root->data)
creat(root->rt, x);
else
creat(root->lt, x);
}
}
void aa(BT *root)
{
if(root)
{
a[++flag] = root->data;
aa(root->lt);
aa(root->rt);
}
}
void bb(BT *root)
{
if(root)
{
b[++flag1] = root->data;
bb(root->lt);
bb(root->rt);
}
}
int main()
{
int n, l, x, i;
BT *root, *root1;
while(~scanf("%d", &n))
{
flag = -1;
root = NULL;
if(n == 0)
break;
scanf("%d", &l);
for(i=0; i<n; i++)
{
scanf("%d", &x);
creat(root, x);
}
aa(root);
while(l--)
{
int k = 1;
flag1 = -1;
root1 = NULL;
for(i=0; i<n; i++)
{
scanf("%d", &x);
creat(root1, x);
}
bb(root1);
for(i=0; i<=flag1; i++)
{
if(a[i] != b[i])
{
k = 0;
break;
}
}
if(k)
printf("Yes\n");
else
printf("No\n");
}
}
return 0;
}