判断t1树是否包含t2树全部的拓扑结构

判断t1树是否包含t2树全部的拓扑结构

题目描述

给定彼此独立的两棵二叉树，判断 t1 树是否包含 t2 树全部的拓扑结构。

设 t1 树的边集为 E1，t2 树的边集为 E2，若 E2 是 E1 的子集，则表示 t1 树包含 t2 树全部的拓扑结构。

输入描述：

第一行输入两个整数 n 和 root，n 表示二叉树 t1 的总节点个数，root 表示二叉树 t1 的根节点。

以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)

第 n+2 行输入两个整数 m 和 root，m 表示二叉树 t2 的总节点个数，root 表示二叉树 t2 的根节点。

以下 m 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)

输出描述：

如果 t1 树包含 t2 树全部的拓扑结构，则输出 “true”，否则输出 “false”。

示例1

输入

10 1
1 2 3
2 4 5
4 8 9
8 0 0
9 0 0
5 10 0
10 0 0
3 6 7
6 0 0
7 0 0
4 2
2 4 5
5 0 0
4 8 0
8 0 0

输出

true

备注：

$1\le n,m\le 2000$

$1\le fa,lch,rch,root\le n$

---

题解：

递归遍历第一棵树中的每个节点，递归判断以当前节点为根节点的子树是否包含树2。

代码：

#include <cstdio>

using namespace std;

const int N = 2010;

struct BST {
    int val, lch, rch;
};

BST bst1[N], bst2[N];

int n, m;
int rt1, rt2;
int fa, lch, rch;

bool check(int root1, int root2) {
    if (!root2) return true;
    if (!root1 || bst1[root1].val != bst2[root2].val) return false;
    return check(bst1[root1].lch, bst2[root2].lch) && check(bst1[root1].rch, bst2[root2].rch);
}

bool contain(int root1, int root2) {
    if (!root2) return true;
    if (!root1) return false;
    return check(root1, root2) || contain(bst1[root1].lch, root2) || contain(bst1[root1].rch, root2);
}

int main(void) {
    scanf("%d%d", &n, &rt1);
    while (n--) {
        scanf("%d%d%d", &fa, &lch, &rch);
        bst1[fa].val = fa;
        bst1[fa].lch = lch;
        bst1[fa].rch = rch;
    }
    scanf("%d%d", &m, &rt2);
    while (m--) {
        scanf("%d%d%d", &fa, &lch, &rch);
        bst2[fa].val = fa;
        bst2[fa].lch = lch;
        bst2[fa].rch = rch;
    }
    puts(contain(rt1, rt2) ? "true" : "false");
    return 0;
}