[ACM] hdu 3791 二叉搜索树

Problem Description

判断两序列是否为同一二叉搜索树序列

Input

开始一个数n，(1<=n<=20) 表示有n个需要判断，n= 0 的时候输入结束。

接下去一行是一个序列，序列长度小于10，包含(0~9)的数字，没有重复数字，根据这个序列可以构造出一颗二叉搜索树。

接下去的n行有n个序列，每个序列格式跟第一个序列一样，请判断这两个序列是否能组成同一颗二叉搜索树。

Output

如果序列相同则输出YES，否则输出NO

Sample Input

2
567432
543267
576342
0

Sample Output

YES
NO

Source

浙大计算机研究生复试上机考试-2010年
 

解题思路：

搜索树的特点为左儿子的数值比父节点小，右儿子的数值比父节点大，根据这个特性插入数值，构造搜索树。

代码：

#include <iostream>
#include <string.h>
using namespace std;
int tr[1024],tr1[1024];//1024的选择要根据元素的个数，极限情况下是所有的节点只有右节点，则节点编号为1->1*2+1(3)->7->15....，每个节点对应一个元素
char s[12],s1[12];

void insert(int num[],int index,int x)//，数组的编号为树中节点的编号，向二叉树中插入元素
{
if(num[index]==-1)
num[index]=x;
else if(num[index]<x)
insert(num,index*2,x);
else
insert(num,index*2+1,x);
}
int main()
{
int n;
while(cin>>n&&n)
{
memset(tr,-1,sizeof(tr));
cin>>s;
for(int i=0;i<strlen(s);i++)
{
int temp=s[i]-'0';
insert(tr,1,temp);
}//构造主树
while(n--)
{
memset(tr1,-1,sizeof(tr1));
cin>>s1;
for(int i=0;i<strlen(s1);i++)
{
int temp=s1[i]-'0';
insert(tr1,1,temp);
}//构造被判断的树
int flag=0;
for(int i=0;i<1024;i++)
{
if(tr[i]!=tr1[i])
{
flag=1;
break;
}
}
if(flag==1)
cout<<"NO"<<endl;
else
cout<<"YES"<<endl;
}
}
return 0;
}