头歌——基于二叉链表的树结构相等的判断

任务描述

设二叉树中每个结点的元素均为一个字符，按先序遍历的顺序建立二叉链表，按此方法创建两棵二叉树，然后编写递归算法判断这两棵树是否相等。

编程要求

输入

多组数据，每组数据有两行。每行为一个二叉树的先序序列（序列中元素为‘0’时，表示该结点为空）。当输入只有一个“0”时，输入结束。

输出

每组数据输出一行。若两个二叉树相等输出“YES”，否则输出“NO”。

测试说明

平台会对你编写的代码进行测试：

测试输入： abcd00e00f00ig00h00 abcd00e00f00ig00h00 abd00e00cf00g00 abd00e00cf00h00 0

预期输出： YES NO

#define _CRT_SECURE_NO_WARNINGS 1


#include<iostream>
using namespace std;


//二叉树的二叉链表存储表示
typedef struct BiTNode
{
    char data;
    struct BiTNode* lchild, * rchild;
}BiTNode, * BiTree;

//按先序次序输入二叉树结点的值，采用递归方法创建二叉树
void CreateBiTree(BiTree& T, char S[], int& i)
{
    
    if (S[i] == '0')
    {
        T = NULL;
    }
    else
    {
        T = new BiTNode;
        T->data = S[i];
        CreateBiTree(T->lchild, S, ++i);
        CreateBiTree(T->rchild, S, ++i);
    }
}


/*这里不可以使用cin >> S[i]; 书上的算法和实际代码是不一样的，在main()函数中
代码给了cin >> S1，所以函数中不可以使用cin,
这段代码放在else外面，if上面也是正确的。
T = new BiTNode;
T->data = S[i];

*/
//void CreateBiTree(BiTree& T, char S[], int& i)
//{
//    cin >> S[i];
//    
//
//    if (S[i] == '0')
//    {
//        T = NULL;
//    }
//    else
//    {
//        T = new BiTNode;
//        T->data = S[i];
//        CreateBiTree(T->lchild, S, ++i);
//        CreateBiTree(T->rchild, S, ++i);
//    }
//}


int Compare(BiTree T1, BiTree T2)
{

    /*如果满足第一个if语句的条件，那么剩下的if语句都将被跳过，不
    会再执行。这是因为在满足第一个if语句条件后，程序会执行对应的
    代码块，并直接跳出--{整个}--if-else if-else结构，继续执行后续
    的代码，因此，后面的if语句将不会被判断和执行。*/


    if (T1 == NULL && T2 == NULL) // 两棵树都为空，认为相等
        return 1;

    if (T1 == NULL || T2 == NULL) // 两棵树中仅有一棵为空，认为不相等
        return 0;

    if (T1->data != T2->data) // 节点值不相等，认为不相等
        return 0;

    /*这是错误的
    if (T1->data == T2->data) 
        return 1;*/

    /*这里要注意，当该结点的值存在且相等时，是接着递归左右子树，不可以
    进行判断，返回1，可以理解为结点的值存在且相等，是进行左右子树递归判断
    的一个条件*/

    int leftEqual = Compare(T1->lchild, T2->lchild); // 比较左子树
    int rightEqual = Compare(T1->rchild, T2->rchild); // 比较右子树

    return leftEqual && rightEqual; // 左右子树都相等才认为相等
}

int main()
{
    char S1[100], S2[100];


    /*，循环读取输入的字符数组 S1 和 S2，直到输入以 '0' 开头的字符串结
    束循环。每次循环创建两棵二叉树，并将其传入 Compare 函数进行比较。根
    据比较结果输出相应的结果。*/

    while (cin >> S1 && S1[0] != '0')
    {
        cin >> S2;

        int i = -1, j = -1;

        BiTree T1, T2;

        CreateBiTree(T1, S1, ++i);
        CreateBiTree(T2, S2, ++j);
        
        if (!Compare(T1, T2))
            cout << "NO" << endl;
        else
            cout << "YES" << endl;
    }
    return 0;
}