问题 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;
}