PTA 7-3 树的同构 (25分)

题目

题目链接：https://pintia.cn/problem-sets/15/problems/711
给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2，则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的，因为我们把其中一棵树的结点A、B、G的左右孩子互换后，就得到另外一棵树。而图2就不是同构的。

思路解析

树的同构，其实是两个树的相同结点的左右孩子名字相同，位置可能不同。
所以，问题转化成判断每个结点的孩子。
因此单独写一个函数用于判断两个结点的孩子结点。

代码

#include <stdio.h>
#include <stdlib.h>

typedef struct {
    char name;//结点名称
    int left;//左子树
    int right;//右子树
}Tree;

Tree tree1[11], tree2[11];//两棵树
int N1, N2;//两棵树的结点数

void input(Tree *tree, int N)
{
    int i;
    char c;
    for(i = 0; i < N; i++)
    {
        scanf("%c", &(tree[i].name));
        //输入左子树
        scanf(" %c ", &c);
        if(c=='-')
            tree[i].left = -1;
        else
            tree[i].left = c - '1' + 1;
        //输入右子树
        scanf("%c", &c);
        if(c=='-')
            tree[i].right = -1;
        else
            tree[i].right = c - '1' + 1;
        getchar();//吃掉回车
    }
}

int search(int no, char name)//根据名字返回位置
{
    int i;
    if(no == 1)
    {
        for(i=0; i<N1; i++)
            if(tree1[i].name == name)
                return i;
    }
    else if(no == 2)
    {
        for(i=0; i<N2; i++)
            if(tree2[i].name == name)
                return i;
    }
    //没找到，返回错误
    return -1;
}

int compare_lr(int a, int b)//判断两棵树的某结点的子孩子是否一致
{//一致返回1，不同返回0
 //a是第一棵树的结点，b是第二棵树的结点
    if(tree1[a].left == -1 && tree1[a].right == -1)
    {//第一个树的结点是根节点
        if(tree2[b].left == -1 && tree2[b].right == -1)
        //第二个树的结点是根节点
            return 1;
        else
            return 0;
    }
    int m, n;
    if((m = tree1[a].left) == -1 || (m = tree1[a].right) == -1)
    {//第一个结点只有一个节点
        if((n = tree2[b].left) == -1 || (n = tree2[b].right) == -1)
        {//如果第二个结点也只有一个节点
            if(tree1[m].name == tree2[n].name)
                return 1;
            else
                return 0;
        }
        else//如果孩子数不同，错误
            return 0;
    }
    if(tree2[b].left != -1 && tree2[b].right != -1)
    {//第一个结点肯定有两个节点
     //判断第二个结点有没有两个结点
        //保存孩子结点的名称
        char First_Left = tree1[tree1[a].left].name;
        char First_Right = tree1[tree1[a].right].name;
        char Second_Left = tree2[tree2[b].left].name;
        char Second_Right = tree2[tree2[b].right].name;

        if(First_Left == Second_Left)
        {//如果第一个结点的左孩子等于第二个结点的左孩子
            if(First_Right == Second_Right)
                return 1;
            else
                return 0;
        }
        else if(First_Left == Second_Right)
        {//如果第一个结点的左孩子等于第二个结点的右孩子
            if(First_Right == Second_Left)
                return 1;
            else
                return 0;
        }
        else//第一个结点左孩子不在第二个结点的孩子里
            return 0;
    }
    else//如果孩子数不同，错误
        return 0;
}

int compare()//判断两棵树是不是同构树
{
    //遍历第一棵树，并判断每个结点对应的子孩子是否相同
    int i;
    int index_B;
    for(i = 0; i < N1; i++)
    {
        //查找1树结点名字对应的2树结点
        index_B = search(2, tree1[i].name);
        //判断结点子孩子是否一致
        if(!compare_lr(i, index_B))//一旦对不上，返回0
            return 0;
    }

    return 1;
}

int main()
{
    scanf("%d", &N1);
    getchar();//吞掉回车
    input(tree1, N1);//输入第一棵树
    scanf("%d", &N2);
    getchar();
    input(tree2, N2);//输入第二棵树
    if(N1 != N2)
        printf("No\n");
    else 
        if(compare())//判断两棵树是不是同构树
            printf("Yes\n");
        else
            printf("No\n");

    return 0;
}

后记

我的代码也许不如别人的短，但一定容易看明白

---

用于方便输入输出的代码

int main()
{
	 freopen("sample.in", "r", stdin);
     freopen("sample.out", "w", stdout);
             
     /* 同控制台输入输出 */
             
     fclose(stdin);
     fclose(stdout);
	 return 0;
}