14-动态查找与查找二叉树-
题目描述
二叉树分别以数组存储方式创建、以先序遍历序列创建。输入二叉树的数组存储、先序遍历结果,判断根据它们创建的二叉树是否是同一棵二叉树。
输入
测试次数t
每组测试数据两行:
第一行:二叉树的数组存储(英文字母表示树结点,#表示空树)
第二行:二叉树的先序遍历结果(英文字母表示树结点,#表示空树)
输出
对每组测试数据,如果两种方式创建的是同一棵二叉树,输出YES,否则,输出NO。
输入样例
3
ABCDE
ABD##E##C
ABC##DE####W##F
AB##CDW###E#F##
abc##d
ab##c#d
输出样例
YES
YES
NO
#include<iostream>
#include<queue>
using namespace std;
struct node
{
char data;
struct node *lchild;
struct node *rchild;
};
class BiTree
{
node *root;
string a;//二叉树的数组存储
string ch;//二叉树的先序遍历结果
int len;//数组存储长度
int i=-1;
void PreOrder(node *&p)//还原先序遍历结果
{
if(i==ch.length())
return;
i++;
if(ch[i]!='#')
{
p=new node();
p->data=ch[i];
PreOrder(p->lchild);
PreOrder(p->rchild);
}
else
{
p=new node();
p->data='#';
p->lchild=NULL;
p->rchild=NULL;
}
}
public:
BiTree()
{
cin>>a>>ch;
len=a.length();
}
void PreOrder()
{PreOrder(root);}
void BFSTraverse()
{
node *p;
queue<node*> Q;
Q.push(root);
for(int i=0;i<len;i++)
{
p=Q.front();
Q.pop();//一个一个字符比较
if(p!=0)
{
if(p->data!=a[i])
{
cout<<"NO"<<endl;
return;//结束函数
}
Q.push(p->lchild);
Q.push(p->rchild);
}
}
cout<<"YES"<<endl;
}
};
int main()
{
int t;
cin>>t;
while(t--)
{
BiTree B;
B.PreOrder();
B.BFSTraverse();
}
return 0;
}