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

 

搜索树的特点为左儿子的数值比父节点小，右儿子的数值比父节点大，根据这个特性插入数值，构造搜索树。

#include <iostream>
#include <string.h>
#include <iomanip>
#include <algorithm>
#include <map>
#include <stdio.h>
using namespace std;
int str1[1024],str2[1024];
void insert(int num[],int index,int x)
{
    if(num[index]==-1)
        num[index]=x;
    else if(num[index]<x)
        insert(num,2*index,x);
    else
        insert(num,2*index+1,x);
}
int main()
{
    int n;
    char s[10];
    while(cin>>n&&n!=0)
    {
        memset(str1,-1,sizeof(str1));
        cin>>s;
        for(int i=0;i<(int)strlen(s);i++)
            insert(str1,1,s[i]-'0');
        while(n--)
        {
            memset(str2,-1,sizeof(str2));
            cin>>s;
            for(int i=0; i<(int)strlen(s); i++)
                insert(str2,1,s[i]-'0');
            int flag=1;
            for(int i=0;i<1024;i++)
                if(str1[i]!=str2[i])
                {
                   flag=0;
                   break;
                }
            if(flag==1)
                cout<<"YES"<<endl;
            else
                cout<<"NO"<<endl;
        }
    }
}