[ACM] 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

Source

浙大计算机研究生复试上机考试-2010年

 

解题思路：

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

代码：

#include <iostream>
#include <string.h>
using namespace std;
int tr[1024],tr1[1024];//1024的选择要根据元素的个数，极限情况下是所有的节点只有右节点，则节点编号为1->1*2+1(3)->7->15....，每个节点对应一个元素
char s[12],s1[12];

void insert(int num[],int index,int x)//，数组的编号为树中节点的编号，向二叉树中插入元素
{
    if(num[index]==-1)
        num[index]=x;
    else if(num[index]<x)
        insert(num,index*2,x);
    else
        insert(num,index*2+1,x);
}
int main()
{
    int n;
    while(cin>>n&&n)
    {
        memset(tr,-1,sizeof(tr));
        cin>>s;
        for(int i=0;i<strlen(s);i++)
        {
            int temp=s[i]-'0';
            insert(tr,1,temp);
        }//构造主树
        while(n--)
        {
           memset(tr1,-1,sizeof(tr1));
            cin>>s1;
            for(int i=0;i<strlen(s1);i++)
            {
                int temp=s1[i]-'0';
                insert(tr1,1,temp);
            }//构造被判断的树
            int flag=0;
            for(int i=0;i<1024;i++)
            {
                if(tr[i]!=tr1[i])
                {
                    flag=1;
                    break;
                }
            }
            if(flag==1)
                cout<<"NO"<<endl;
            else
                cout<<"YES"<<endl;
        }
    }
    return 0;
}