1920 Problem B 二叉搜索树

问题 B: 二叉搜索树

时间限制: 1 Sec  内存限制: 32 MB
提交: 52  解决: 36
 

题目描述

判断两序列是否为同一二叉搜索树序列

输入

开始一个数n，(1<=n<=20) 表示有n个需要判断，n= 0 的时候输入结束。
接下去一行是一个序列，序列长度小于10，包含(0~9)的数字，没有重复数字，根据这个序列可以构造出一颗二叉搜索树。
接下去的n行有n个序列，每个序列格式跟第一个序列一样，请判断这两个序列是否能组成同一颗二叉搜索树。

输出

如果序列相同则输出YES，否则输出NO

样例输入

6
45021
12045
54120
45021
45012
21054
50412
0

样例输出

NO
NO
YES
NO
NO
NO

经验总结

首先，建立最原始的二叉排序树，然后，存储其先序和中序遍历的序列，接下来的n个例子，每个都建立二叉排序树，并且进行先序与中序遍历，只有这两个遍历的序列完全相等，才可以说明这两个二叉排序树完全相同，然后输出就行啦！~~( ｡ớ ₃ờ)ھ

正确代码

#include <cstdio>
#include <vector>
using namespace std;

struct node
{
	int data;
	node *lchild;
	node *rchild;
};
void insert(node * &root, int x)
{
	if(root==NULL)
	{
		node * temp=new node;
		temp->data=x;
		temp->rchild=temp->lchild=NULL;
		root=temp;
		return ;
	}
	if(x==root->data)
	{
		return;
	}
	else if(x>root->data)
	{
		insert(root->rchild,x);
	}
	else
	{
		insert(root->lchild,x);
	}
}
void pre(node *root,vector<int> &answer)
{
 	if(root==NULL)
 	{
 		return ;
	}
	answer.push_back(root->data);
	pre(root->lchild,answer);
	pre(root->rchild,answer);
}
void in(node *root,vector<int> &answer)
{
 	if(root==NULL)
 	{
 		return ;
	}
	in(root->lchild,answer);
	answer.push_back(root->data);
	in(root->rchild,answer);
}
int main()
{
	int n,data;
	char temp[15];
	vector<int> opre,oin,tpre,tin;
    while(~scanf("%d",&n))
    {
    	if(n==0)
    		break;
    	opre.clear();
    	oin.clear();
    	scanf("%s",temp);
    	int k=0;
    	while(temp[k]!='\0')
    		k++;
    	node * root=NULL;
    	for(int i=0;i<k;++i)
    	{
			insert(root,temp[i]-'0');
		}
		pre(root,opre);
		in(root,oin);
		for(int i=0;i<n;++i)
		{
			tpre.clear();
			tin.clear();
			scanf("%s",temp);
	    	k=0;
	    	while(temp[k]!='\0')
	    		k++;
	    	root=NULL;
	    	for(int i=0;i<k;++i)
	    	{
				insert(root,temp[i]-'0');
			}
			pre(root,tpre);
			in(root,tin);
			if(opre==tpre&&oin==tin)
				printf("YES\n");
			else
				printf("NO\n");
		}
	}
    return 0;
}