树的子结构

一、需求

• 输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)B是A的子结构， 即 A中有出现和B相同的结构和节点值。
• 例如，给定树A：

给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B：

   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

二、思路分析

1. 写一个dfs函数，用于判断某一节点下的子树是否与B相同；
2. 在isSubstructure函数中遍历树A中的所有节点，只要有一个节点满足步骤1中的情况；

三、代码实现

class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null) return false;
        return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
    //判断以某一节点为根节点的子树是否与B相同
    public boolean dfs(TreeNode node, TreeNode B) {
        //问题：为什么下面两行代码的位置不能互换?
        //答:假设现在树A和树B各有1个节点，节点值为3，那么在判断dfs(node.left, B.left)时，
        //会出现问题。
        if(B == null) return true;
        if(node == null) return false;
        return node.val == B.val && dfs(node.left, B.left) && dfs(node.right, B.right);
    }
}

四、复杂度分析

• 时间复杂度为O(MN)，M和N分别是树A和树B的结点数量，寻找与B相对应的根结点耗费的时间复杂度为O(M)，每次dfs耗费的时间复杂度为O(N)，因此总的时间复杂度为O(MN)；
• 空间复杂度为O(M)，递归的调用深度最大为A树的结点数量M。