HDU 3791 二叉搜索树

二叉搜索树

[align=left]Problem Description[/align]
判断两序列是否为同一二叉搜索树序列
 

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

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

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

[align=left]Output[/align]
如果序列相同则输出YES，否则输出NO

 

[align=left]Sample Input[/align]

2
567432
543267
576342
0

 

[align=left]Sample Output[/align]

YES
NO

分析：二叉搜索树求解，用数组存储节点，然后两数组相比就行了，由于树的深度最大为10，故数组大小设置为 （2^10）

AC代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
const int maxn=1<<12;
int a[maxn];
int b[maxn];

void insert(int* t,char c,int deep){
int c1=c-'0';
if(t[deep]==-1){
t[deep]=c1;
}
else if(c1<t[deep]){
insert(t,c,deep<<1);
}
else insert(t,c,deep<<1|1);
}

int main(){
int T;
while(scanf("%d",&T)==1 &&T){
memset(a,-1,sizeof(a));
char s[12];
scanf("%s",s);
for(int i=0;i<strlen(s);i++){
insert(a,s[i],1);
}

while(T--){
memset(b,-1,sizeof(b));
scanf("%s",s);
for(int i=0;i<strlen(s);i++){
insert(b,s[i],1);
}

int i;
for(i=0;i<=(1<<10);i++)
{
if(a[i]!=b[i])break;
}

if(i>(1<<10))printf("YES\n");
else printf("NO\n");
}
}
return 0;
}