HDU - 3791 二叉搜索树

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=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

这破题目…丢死人了，因为0也在序列里面所以初始化要比0小，还有输入是多次的，一开始看清题目就不会错这么多次了…😥

代码如下：

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 10 + 5;
int n, len, len2, a[maxn], b[maxn], L[maxn], R[maxn], L2[maxn], R2[maxn];
bool dfs(int pos) {
    if (L[pos] != L2[pos] || R[pos] != R2[pos])return false;
    if (L[pos] != -1) {
        if (R[pos] != -1)return dfs(L[pos]) && dfs(R[pos]);
        else return dfs(L[pos]);
    }
    if (R[pos] != -1)return dfs(R[pos]);
    return true;
}
int main(void) {
    while (~scanf("%d", &n) && n) {
        getchar();
        len = 0;
        memset(L, -1, sizeof(L));
        memset(R, -1, sizeof(R));
        char ch = getchar();
        while (ch != '\n') {
            a[len++] = ch - '0';
            ch = getchar();
        }
        int root = a[0];
        for (int i = 1; i < len; i++) {
            int val = root;
            while (1) {
                if (val > a[i]) {
                    if (L[val] != -1)val = L[val];
                    else {
                        L[val] = a[i]; break;
                    }
                }
                else {
                    if (R[val] != -1)val = R[val];
                    else {
                        R[val] = a[i]; break;
                    }
                }
            }
        }
        while (n--) {
            memset(L2, -1, sizeof(L2));
            memset(R2, -1, sizeof(R2));
            len2 = 0;
            char ch = getchar();
            while (ch != '\n') {
                b[len2++] = ch - '0';
                ch = getchar();
            }
            if (len != len2) {
                printf("NO\n");
                continue;
            }
            for (int i = 1; i < len; i++) {
                int val = b[0];
                while (1) {
                    if (val > b[i]) {
                        if (L2[val] != -1)val = L2[val];
                        else {
                            L2[val] = b[i]; break;
                        }
                    }
                    else {
                        if (R2[val] != -1)val = R2[val];
                        else {
                            R2[val] = b[i]; break;
                        }
                    }
                }
            }
            bool flag;
            if (a[0] != b[0])flag = false;
            else flag = dfs(root);
            printf("%s", flag ? "YES\n" : "NO\n");
        }
    }
    return 0;
}