二叉搜索树 HDU - 3791

问题：

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

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

思路：

二叉搜索树是按照比根节点小的插入到根结点的左边，比根节点大的插入到右边，然后我们可以记录节点的插入路径，加设根节点路径是零，向左插入路径就是就是当前路径+“0”，向右就+“1”，然后比较每个点的路径是否相同。

代码：

#define N 15
#include<map>
#include<string>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
map<int,string>m,h;
map<string,int>w,q;
int s[N],c[N],n,l,b;
char ss[N],cc[N];
int dfs1(string t)
{
    if(w.find(t)==w.end())
    {
        w[t]=b;//保留t路径的节点b
        m[b]=t;//保留b的路径t
        return 1;
    }
    if(w[t]>b)
        if(dfs1(t+'0'))
            return 1;
    if(w[t]<b)
        if(dfs1(t+'1'))
            return 1;
}
int dfs2(string t)
{
    if(q.find(t)==q.end())
    {
        q[t]=b;
        h[b]=t;
        return 1;
    }
    if(q[t]>b)
        if(dfs2(t+'0'))
            return 1;
    if(q[t]<b)
        if(dfs2(t+'1'))
            return 1;
}
int main()
{
    while(scanf("%d",&n)&&n)
    {
        m.clear();
        w.clear();
        scanf("%s",ss);
        for(int i=0; ss[i]!='\0'; i++)
            s[i]=ss[i]-'0';
        l=strlen(ss);
        m[s[0]]="0";
        w["0"]=s[0];
        for(int i=1; i<l; i++)
        {
            b=s[i];
            dfs1("0");
        }
        sort(s,s+l);
        for(int i=0; i<n; i++)
        {
            scanf("%s",cc);
            h.clear();
            q.clear();
            for(int j=0; cc[j]!='\0'; j++)
                c[j]=cc[j]-'0';
            h[c[0]]="0";
            q["0"]=c[0];
            for(int j=1; j<l; j++)
            {
                b=c[j];
                dfs2("0");
            }
            int flag=0;
            sort(c,c+l);
            for(int j=0; j<l; j++)
                if(s[j]!=c[j]||m[s[j]]!=h[c[j]])
                    flag=1;
            if(flag)
                printf("NO\n");
            else
                printf("YES\n");
        }
    }
    return 0;
}