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
  
  
一.总结
  
  

   
   用结构来写二叉树真的好麻烦，老是各种错误，用数组就来的简单的多，如果数据范围允许的话，用数组来保存二叉树是很理想的
  
  
  
  

   
   
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn=1024;
int tree1[maxn];
int tree2[maxn];
string a,b;
int find(int n,int p,int *tree)
{
    if(tree[p]==-1) return p;
    else if(n>tree[p]) return find(n,2*p+1,tree);
    else return find(n,2*p,tree);
}
void init1()
{
    memset(tree1,-1,sizeof(tree1));
    for(int i=0;i<a.length();i++)
    {
        int p=find(a[i]-48,1,tree1);
        tree1[p]=a[i]-48;
    }
}
void init2()
{
    memset(tree2,-1,sizeof(tree2));
    for(int i=0;i<b.length();i++)
    {
        int p=find(b[i]-48,1,tree2);
        tree2[p]=b[i]-48;
    }
}
int main()
{
    //freopen("/Users/zhangjiatao/Documents/暑期训练/input.txt","r",stdin);
    int n;
    while(scanf("%d",&n)==1)
    {
        if(n==0) break;
        cin>>a;
        init1();
        for(int i=1;i<=n;i++)
        {
            cin>>b;
            int flag=1;
            init2();
            for(int i=1;i<maxn;i++)
            {
                if(tree1[i]!=tree2[i]) {flag=0;break;}
            }
            if(flag==1) cout<<"YES"<<endl;
            else cout<<"NO"<<endl;
        }
    }
    return 0;
}