二叉搜索树
判断两序列是否为同一二叉搜索树序列
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
题解:题目的数据量很小,所以直接数组模拟建树,为了比较的方便直接建成完全二叉树,这样就便于比较了,只要是相同位置必有相同的值,否则不是相同的二叉树。
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
char mp[2020],s[2020];
int a[2020],b[2020];
int tree(int m,int k,int *ab)//每次从根节点开始比较
{
if(ab[k]==-1)
return k;
else if(m>ab[k])
return tree(m,2*k+1,ab);//搜索右孩子
else
return tree(m,2*k,ab);//搜索左孩子
}
void lin1()
{
int i,l1=strlen(mp);
memset(a,-1,sizeof(a));
for(i=0; i<l1; i++)
{
int k=tree(mp[i]-48,1,a);
a[k]=mp[i]-48;
}
}
void lin2()
{
int i,l2=strlen(s);
memset(b,-1,sizeof(b));
for(int i=0; i<l2; i++)
{
int k=tree(s[i]-48,1,b);
b[k]=s[i]-48;
}
}
int main()
{
int n;
while(~scanf("%d",&n)&&n)
{
int i,j,flag;
scanf("%s",mp);
lin1();
for(i=1; i<=n; i++)
{
flag=1;
scanf("%s",s);
lin2();
for(j=1; j<2020; j++)
if(a[j]!=b[j])
{
flag=0;
break;
}
if(flag==1)
printf("YES\n");
else
printf("NO\n");
}
}
return 0;
}