传送门:HDU 3791
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
题解:
首先得知道二叉搜索树的性质:若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉搜索树。
那么我们根据二叉搜索树的生成规则先将一开始给你的序列生成一个二叉搜索树用tree[]存储起来,然后输入n个序列,每一个序列都生成一个二叉搜索树建立一个临时数组tree2[]存储,比较两个数组是否相同,相同输出“YES”,不同输出“NO”。
AC代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
char a[25],b[25];
int tree[1005],trie[1005];
int t;
void build(int tree[],int pos,int k)//建一个二叉搜索树
{
if(tree[pos]==-1)//搜到底
{
tree[pos]=k;
return ;
}
if(k<tree[pos]) build(tree,pos*2,k);//比根节点小的都在左边
else build(tree,pos*2+1,k);//比根节点大的都在右边
}
int main()
{
while(scanf("%d",&t)&&t)
{
memset(tree,-1,sizeof(tree));
scanf("%s",a);
int lena=strlen(a);
for(int i=0;i<lena;i++)//建树
{
build(tree,1,a[i]-'0');
}
while(t--)
{
int judge=1;
scanf("%s",b);
memset(trie,-1,sizeof(trie));
int lenb=strlen(b);
for(int i=0;i<lenb;i++)
{
build(trie,1,b[i]-'0');
}
for(int i=1;i<1000;i++)//对比两颗树是否相等
{
if(tree[i]!=trie[i])
judge=0;
}
if(judge) printf("YES\n");
else printf("NO\n");
}
}
return 0;
}