HDOJ 3791 二叉搜索树（二叉排序树的建立和遍历）



http://acm.hdu.edu.cn/showproblem.php?pid=3791

二叉搜索树

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

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
  
  

 

//题意：给出一串数字，可以建立一颗二叉排序树，后边又给出n串数字，
//判断这n串数字建立起来的二叉排序树是否和原来的一致。 
//解题思路：首先用给出的数字串建立一颗二叉排序树，遍历，得到遍历结果，以此为标准; 
//然后对后边给出的每一串数字，建立二叉排序树，按照和上面同样的方法进行遍历，
//如果遍历的结果和标准结果一样，说明建立的二叉排序树和标准的是一样的，输出YES，
//如果不一样，说明建立的和标准的不一样，输出NO 
#include<stdio.h>
#include<string.h>
#include<malloc.h>
struct Node
{
	int date;
	struct Node *lchild,*rchild;
};
int k,l,a[20],b[20];
void CreatBintree(struct Node *root,int *num,int n)
{//建立二叉排序树 
	int i;
	struct Node *p,*head;
	for(i=1;i<n;i++)
	{
		//-----建立新节点-----//
		p=(struct Node *)malloc(sizeof(struct Node));
		p->date=num[i];
		p->lchild=NULL;
		p->rchild=NULL;
		//--------------------//
		head=root;
		while(1)
		{
			if(p->date<head->date&&!head->lchild)
			{
				head->lchild=p;
				break;
			}
			else if(p->date>head->date&&!head->rchild)
	        {
	        	head->rchild=p;
	        	break;
	        }
	        
	        if(p->date<head->date&&head->lchild)
	        head=head->lchild;
	        else if(p->date>head->date&&head->rchild)
	        head=head->rchild;
	    }
	}
}
void Ergodic_1(struct Node *root,int *a)
{
	a[k++]=root->date;
	if(root->lchild) Ergodic_1(root->lchild,a);
	if(root->rchild) Ergodic_1(root->rchild,a);
}
void Ergodic_2(struct Node *root,int *b)
{
	b[l++]=root->date;
	if(root->lchild) Ergodic_2(root->lchild,b);
	if(root->rchild) Ergodic_2(root->rchild,b);
}

//void Delete(struct Node *root)
//{
//	if(root->lchild) Delete(root->lchild);
//	if(root->rchild) Delete(root->rchild);
//	free(root);
//}
int main()
{
	int m,n,i,j;
	char s[20];
	while(scanf("%d",&m)&&m!=0)
	{
		scanf("%s",s);
		n=strlen(s);
		int *num=(int *)malloc(sizeof(int)*n);
		for(i=0;i<n;i++)
		num[i]=s[i]-'0';
		//-----建立根节点-----//
		struct Node *root;
		root=(struct Node *)malloc(sizeof(struct Node));
		root->date=num[0];
		root->lchild=NULL;
		root->rchild=NULL;
		//--------------------//
		
		CreatBintree(root,num,n);//建立二叉排序树 
		k=0;
		Ergodic_1(root,a);
		
		for(i=0;i<m;i++)
		{//对每一次输入的顺序进行判断，看能否构成二叉排序树 
			scanf("%s",s);
			for(j=0;j<n;j++)//将输入的字符串转换成数字存入num数组中 
			num[j]=s[j]-'0';
			//-----建立根节点-----//
		    struct Node *s;
		    s=(struct Node *)malloc(sizeof(struct Node));
		    s->date=num[0];
		    s->lchild=NULL;
		    s->rchild=NULL;
		    //--------------------//
		    CreatBintree(s,num,n);//为要判断的一串数字建立二叉排序树
		    l=0;
		    memset(b,0,sizeof(b));
		    Ergodic_2(s,b);//对要判断的数子建立的二叉树进行判断 
		    
		    int sum=0;
		    for(j=0;j<n;j++)
		    if(a[j]==b[j]) sum+=1;
		    
		    if(sum==n)
		    printf("YES\n");
		    else 
			printf("NO\n");
			
		//	Delete(s);//释放内存 
		}
		//Delete(root); //释放内存 
	}
	return 0;
}