九度 1009：二叉搜索树

九度 1009：二叉搜索树

原题OJ链接：http://ac.jobdu.com/problem.php?pid=1009

题目描述：

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

输入：

开始一个数n，(1<=n<=20) 表示有n个需要判断，n= 0 的时候输入结束。
接下去一行是一个序列，序列长度小于10，包含(0~9)的数字，没有重复数字，根据这个序列可以构造出一颗二叉搜索树。
接下去的n行有n个序列，每个序列格式跟第一个序列一样，请判断这两个序列是否能组成同一颗二叉搜索树。

输出：

如果序列相同则输出YES，否则输出NO

样例输入：

2
567432
543267
576342
0

样例输出：

YES
NO

解题思路：

包括中序遍历在内的两种遍历结果（前序+中序/后序+中序）可以唯一确定一棵二叉树。只需对两棵树进行包含中序遍历在内的两种遍历，若两种遍历的结果都相同，那么就可以判断这两棵树是完全相同的。
本题采用的是后序+中序的方式。

源代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;

struct Node{
    Node *lchild;
    Node *rchild;
    int c;
} Tree[110];

int loc;

Node *creat(){
    Tree[loc].lchild=Tree[loc].rchild=NULL;
    return &Tree[loc++];
}

char str1[25],str2[25];
/*
保存二叉排序树的遍历结果，将每一棵树的后序遍历得到的
字符串与中序遍历得到的字符串连接，得到遍历结果字符串
*/
int size1,size2;//保存在字符数组中的遍历得到的字符个数
char *str;//当前正在保存字符串
int *size;//当前正在保存字符串中字符个数

void postOrder(Node *T){
    if(T->lchild!=NULL){
        postOrder(T->lchild);
    }
    if(T->rchild!=NULL){
        postOrder(T->rchild);
    }
    str[(*size)++]=T->c+'0';
}

void inOrder(Node *T){
    if(T->lchild!=NULL){
        inOrder(T->lchild);
    }
    str[(*size)++]=T->c+'0';
    if(T->rchild!=NULL){
        inOrder(T->rchild);
    }
}

Node *Insert(Node *T,int x){
    if(T==NULL){
        T=creat();
        T->c=x;
        return T;
    }
    else if(x<T->c){
        T->lchild=Insert(T->lchild,x);
    }
    else if(x>T->c){
        T->rchild=Insert(T->rchild,x);
    }
    return T;
}

int main(){
    int n;
    //char tmp[12];
    string tmp;
    while(scanf("%d",&n)!=EOF && n!=0){
        loc=0;
        Node *T=NULL;
        //scanf("%s",tmp);
        cin>>tmp;
        int len=tmp.length();
        for(int i=0;i<len;i++){
            T=Insert(T,tmp[i]-'0');
        }
        size1=0; 
        str=str1;//将正在保存的字符串设定为第一个字符串
        size=&size1;//将正在保存字符串中的字符个数指针指向size1
        postOrder(T);//后序遍历
        inOrder(T);//中序遍历
        str1[size1]=0;//向第一个字符串的最后一个字符后添加空字符，方便实用字符串函数


        while(n--!=0){
            cin>>tmp;
            Node *T2=NULL;
            for(int i=0;i<len;i++){
                T2=Insert(T2,tmp[i]-'0');
            }
            size2=0;
            str=str2;
            size=&size2;
            postOrder(T2);
            inOrder(T2);
            str2[size2]=0;
            puts(strcmp(str1,str2)==0?"YES":"NO");//两个字符串相等返回0
        }
    }
    return 0;
}

代码引自《王道机试指南》