【HDU No. 3791】 二叉搜索树
【题意】
判断两个序列是否为同一个二叉搜索树序列。
【输入输出】
输入:
第1行包含一个数n (1≤n ≤20),表示有n 个序列需要判断,在n =0时输入结束。
接下来的一行是一个序列,序列长度小于10,包含0~9的数字,没有重复的数字,根据这个序列可以构造出一棵二叉搜索树。再接下来的n 行有n 个序列,每个序列的格式都跟第1个序列一样,请判断这两个序列能否组成同一棵二叉搜索树。
输出:
如果序列相同,则输出“YES”,否则输出“NO”。
【样例】
【思路分析】
这道题可以通过判断二叉搜索树的先序遍历序列是否相同来输出结果。因为根据二叉搜索树的中序有序性,先序遍历序列相同,中序遍历序列也一定相同。通过先序遍历序列和中序遍历序列可以唯一确定一棵二叉树。
【算法设计】
① 使用二叉搜索树,先将每个数字都存进二叉搜索树中,得到先序遍历;
② 将后面每一行的每个数字都存进二叉搜索树中,得到先序遍历,比较其是否相等,如果相等,则输出“YES”,否则输出“NO”。
【算法实现】
#include<cstdio>
#include<cstring>
using namespace std;
int cnt;
char a[15] , b[15] , c[15];
typedef struct node{
int num;
node *lc , *rc;
}*nodeptr;
nodeptr root;
void Insert(nodeptr &t, int x){ //将x 插入到二叉搜索树t 中
nodeptr p;
if(t == NULL){
p = new node;
p->lc = NULL;
p->rc = NULL;
p->num = x;
t = p;
}
else{
if(x <= t->num){
Insert(t->lc , x); //插入到左子树
}else{
Insert(t->rc , x); //插入到右子树
}
}
}
void preorder(nodeptr t , char b[]){ //中序遍历
if(t){
b[cnt++] = t->num + '0';
preorder(t->lc , b);
preorder(t->rc , b);
}
}
int main(){
int n;
while(scanf("%d", &n) , n){
cnt = 0;
root = NULL;
scanf("%s" , a);
for(int i = 0 ; a[i] != '\0' ; i++){
Insert(root , a[i] - '0');
}
preorder(root , b);
b[cnt] = '\0';
while(n --){
cnt = 0;
root = NULL;
scanf("%s" , a);
for(int i = 0 ; a[i] != '\0' ; i++){
Insert(root , a[i] - '0');
}
preorder(root , c);
c[cnt] = '\0';
if(strcmp(b , c) == 0){ //c++字符串可以用==比较
printf("YES\n");
}else{
printf("NO\n");
}
}
}
return 0;
}