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
  
  

分析：二叉搜索树求解，用数组存储节点，然后两数组相比就行了，由于树的深度最大为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;
}