HDU 3791 二叉搜索树

二叉搜索树

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2566    Accepted Submission(s): 1117

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 <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>

const int MAX = 1025;

char tra[MAX],trb[MAX],seq[11];

bool Insert(char tr[],char c){
	int st = 1;
	bool mark = true;

	while(1){
		if(tr[st]==0){
			tr[st] = c;
			if(tr[st]!=tra[st])mark = false;
			break;
		}
		if(tr[st]<c){
			st = (st<<1)+1;
		}else{
			st = (st<<1);
		}
	}
	return mark;
}

bool BuildTree(char tr[],char seq[],int len){
	int i;
	bool mark = true;

	for(i=0;i<len;++i){
		mark = Insert(tr,seq[i]);
		if(!mark)break;
	}
	return mark;
}

int main(){
	int n,len;
	bool mark;

	//freopen("in.txt","r",stdin);
	while(scanf("%d",&n) && n){
		memset(tra,0,sizeof(tra));
		scanf("%s",seq);
		len = strlen(seq);
		BuildTree(tra,seq,len);
		while(n--){
			scanf("%s",seq);
			len = strlen(seq);
			mark = true;
			memset(trb,0,sizeof(trb));
			mark = BuildTree(trb,seq,len);
			if(mark){
				printf("YES\n");
			}else{
				printf("NO\n");
			}
		}
	}
	return 0;
}