hdu 3791 二叉搜索树

二叉搜索树

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

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>
using namespace std;
struct Node{//树节点 
	int v;
	Node *l,*r;
	Node():l(NULL),r(NULL){}
	Node(int val):v(val),l(NULL),r(NULL){}
};
typedef Node* tree;
string s;
int N;
tree & search(tree& t,int &e){//递归式的搜寻插入节点位置的引用 （注意这里的引用） 
	if(t==NULL)return t;
	return search(e<t->v?t->l:t->r,e); 
}
tree build(string&s){//建树 
	tree t=new Node;
	t->v=(int)(s[0]-'0');
	for(int i=1;i<s.size();i++){
		int n=s[i]-'0';
		search(t,n)=new Node(n);//得益于search的返回值为引用类型，此处得以简洁的实现 
	}
	return t;		
}
bool cmp(tree t1,tree t2){//比较两棵树 
	if(t1==NULL){
		if(t2==NULL)return true;
		else return false;
	}
	else if(t2==NULL)return false;
	if(t1->v!=t2->v)return false;
	else return cmp(t1->l,t2->l)&&cmp(t1->r,t2->r);
}
void del(tree &t){//归还系统内存 
	if(t->l)del(t->l);
	if(t->r)del(t->r);
	delete t;
}
int main(){
	int i,j,k,m,n;
	while(cin>>N){
		if(!N)return 0;
		cin>>s;
		tree mo=build(s);//模式树 
		for(i=1;i<=N;i++){
			cin>>s;
			tree t=build(s);
			if(cmp(t,mo))cout<<"YES\n";
			else cout<<"NO\n";
			del(t);
		}
		del(mo);
	}
}