二叉搜索树

二叉搜索树

 

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

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