【JZ-26】树的子结构（树、递归）

题目

方法一-递归

参考

算法思路

若树 B 是树 A 的子结构，则子结构的根节点一定是树 A 中的某一个节点，假设该节点为 $n_a$，那么整个判断过程可以分为两步：

1. 前序遍历树 A 的每一个节点 $n_a$（对应函数 $recur(A,B)$）
2. 判断树 A 中以 $n_a$ 为根节点的子树是否包含树 B（对应函数 $isSubStructure(A,B)$ ）

函数 $recur(A,B)$

终止条件：

1. 节点 B 为 $null$：说明树 B 已经匹配完成，返回 $true$
2. 节点 A 为 $null$：匹配失败，返回 $false$
3. 节点 A 和节点 B 的值不等：匹配失败，返回 $false$

返回值：

1. 遍历到左子树：$recur(A.left,B.left)$
2. 遍历到右子树：$recur(A.right,B.right)$

函数 $isSubStructure(A,B)$

终止条件：

1. 树 A 或树 B 为 $null$：返回 $false$

返回值：

1. 以 A 为根节点的子树包含树 B：$recur(A,B)$
2. 树 B 是树 A 的左子树的子结构：$isSubStructure(A.left,B)$
3. 树 B 是树 A 的右子树的子结构：$isSubStructure(A.right,B)$

具体代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //判断树 A 是否包含树 B
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null)return false;
        return recur(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
    //前序遍历比较两树对应节点是否相等
    boolean recur(TreeNode A, TreeNode B){
        if(B == null)return true;
        if(A == null || A.val != B.val)return false;
        return recur(A.left, B.left) && recur(A.right, B.right);
    }
}

复杂度分析

• 时间复杂度：$O(mn)$，m、n分别为树 A 和树 B 的节点数。
• 空间复杂度：$O(m)$，额外空间开销取决于递归深度，最坏情况下，树退化为链表，一直遍历到树 A 的叶节点。