剑指offer学习笔记 | 树的子结构

题目

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

示例：

一个给定的树A：

     3
    / \
   4   5
  / \
 1   2

一个给定的树B：

   4 
  /
 1

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

示例

输入：A = [1,2,3], B = [3,1]
输出：false

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

解决方法

1、对特例进行判断。当A和B均为空的时候，返回false。

2、如果A根节点的val值和B根节点的val值相等时，递归判断A和B的左右子树是否相等。否则返回false。

3、递归的结束条件：

返回false：A根节点的val值和B根节点的val值不相等；A为空

返回true：B为空

以上是从根节点开始判断时，B是A的子结构的情形下进行的递归。但是，B不光有可能是A的子结构，也有可能是A左子树的子结构或者右子树的子结构。所以如果从根节点判断B不是A的子结构，还要继续判断B是不是A左子树的子结构和右子树的子结构。

代码实现

public boolean isSubStructure(TreeNode A, TreeNode B) {
    if (A == null || B == null)
        return false;
    //先从根节点判断B是不是A的子结构，如果不是在分别从左右两个子树判断，
    //只要有一个为true，就说明B是A的子结构
    return isSub(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}

boolean isSub(TreeNode A, TreeNode B) {
    //这里如果B为空，说明B已经访问完了，确定是A的子结构
    if (B == null)
        return true;
    //如果B不为空A为空，或者这两个节点值不同，说明B树不是
    //A的子结构，直接返回false
    if (A == null || A.val != B.val)
        return false;
    //当前节点比较完之后还要继续判断左右子节点
    return isSub(A.left, B.left) && isSub(A.right, B.right);
}