描述
设二叉树中每个结点的元素均为一个字符,按先序遍历的顺序建立二叉链表,按此方法创建两棵二叉树,然后编写递归算法判断这两棵树是否相等。
输入
多组数据,每组数据有两行。每行为一个二叉树的先序序列(序列中元素为‘0’时,表示该结点为空)。当输入只有一个“0”时,输入结束。
输出
每组数据输出一行。若两个二叉树相等输出“YES”,否则输出“NO”。
输入样例 1
abcd00e00f00ig00h00
abcd00e00f00ig00h00
abd00e00cf00g00
abd00e00cf00h00
0
输出样例 1
YES
NO
思路(WA的可能情况)
首先遍历算法是一个递归算法,什么是递归?就是重复调用多次,但是每次只执行一遍,所以用的是if(T1&&T2)而不是while
其次,在输出的时候可能会想到在主函数里设置一个flag来接收Traverse的返回值以判断输出信息的思路,但是这个思路在以前或许适用,在这里不适用,因为仔细看,Traverse函数在应该YES的情况根本没有返回值,所以这个方法不适用。
代码:
#include <iostream>
using namespace std;
typedef struct BiTNode
{
char data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
void CreateBiTree(BiTree &T)
{
char ch;
cin>>ch;
if(ch=='0')
T=NULL;
else
{
T=new BiTNode;
T->data=ch;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
}
void CreateBiTree(BiTree &T,char ch)
{
if(ch=='0')
T=NULL;
else
{
T=new BiTNode;
T->data=ch;
CreateBiTree(T->lchild);
CreateBiTree(T->rchild);
}
}
int Traverse(BiTree T1,BiTree T2)
{
if(T1&&T2)//不能写while,因为这是递归,每次执行一次,但是调用多次
{
if(T1->data!=T2->data)
{
cout<<"NO"<<endl;
return -1;//不能用flag,因为这个函数的else不一定返回值 !!
}
else
{
Traverse(T1->lchild,T2->lchild);
Traverse(T1->rchild,T2->rchild);
}
}
}
int main()
{
while(1)
{
char ch;
cin>>ch;
if(ch=='0')
break;
BiTree T1,T2;
CreateBiTree(T1,ch);
CreateBiTree(T2);
if(Traverse(T1,T2)!=-1)
cout<<"YES"<<endl;
}
return 0;
}