【剑指 Offer】 26. 树的子结构

题目描述

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

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

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2

给定的树 B：

   4 
  /
 1

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

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

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

解题思路

    若树B是树A的子结构，则子结构的根节点可能为树A的任意节点。因此，判断树B是否是树A的子结构，需完成以下两步工作：

1. 首先判断A树的根结点与B树的根结点是否相等，相等则继续判断B的左子树（有的话）和A树的左子树是否相等，相等的话在判断B的右子树（有的话）与A树的右子树判断。
2. 若A树的根结点与B树不相等，则先序遍历A树分别从A树的左子树和右子树开始与B树根结点判断。

代码实现

class Solution {
        public boolean isSubStructure(TreeNode A, TreeNode B) {
            if (B == null || A == null) {
                return false;
            }
            /**
             * 根结点相等，则判断左子树和右子树是否相等
             */
            if (A.val == B.val && (helper(A.left, B.left) && helper(A.right, B.right))) {
                return true;
            }
            /**
             * 根结点不相等则递归判断A节点的左子树和右子树是否有B树的子结构
             */
            return isSubStructure(A.left, B) || isSubStructure(A.right, B);
        }

        private boolean helper(TreeNode root1, TreeNode root2) {
        	/**
        	 * 说明B节点已经没有左子树或者右子树了，即A树存在B树子结构
        	 */
            if (root2 == null) {
                return true;
            }
            /**
             * A节点的左子树或右子树已经为空
             */
            if (root1 == null) {
                return false;
            }
            /**
             * A和B左子树或右子树的根结点值也相等，则递归继续判断左子树和右子树。
             */
            if (root1.val == root2.val) {
                return helper(root1.left, root2.left) && helper(root1.right, root2.right);
            } else {
                return false;
            }
        }
    }