LeetCode刷题031 树的子结构

剑指 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

限制：

0 <= 节点个数 <= 10000

题解

首先需要判断A和B的根节点是否为空：A为空，则B肯定不是A的子结构；B为空，空树不是任何树的子结构。然后判断A的根节点与B的根节点值是否相等，若不等，则分别将A树根节点的左右子节点作为根节点，递归调用原函数，分别判断B树是否为A左右子树的子结构。
当A和B根节点的值相同时，我们可以判断B是否包含在A中。若根节点的值相同，则递归调用，分别判断B的左子树是否包含在A的左子树中，B的右子树是否包含在A的右子树中。在递归调用过程中，需要增加这样几个终止条件：当B为空时，则返回true，达到最大深度；当A为空时，此时B不为空，则B肯定不是A的子结构，返回false；当B的根节点值与A的根节点值不相同时，返回false。这里判断的顺序一定要相同，否则当A和B都为空时，会返回false，而不是true。

Code

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null){
            return false;
        }
        boolean flag = false;
        if(A.val == B.val)
            flag = isTreeAHaveTreeB(A, B);
        if(!flag)
            flag = isSubStructure(A.left, B) || isSubStructure(A.right, B);
        return flag;
    }

    public static boolean isTreeAHaveTreeB(TreeNode A, TreeNode B){
        if(B == null)
            return true;
        if(A == null)
            return false;
        if(A.val != B.val)
            return false;
        
        return isTreeAHaveTreeB(A.left, B.left) && isTreeAHaveTreeB(A.right, B.right);
    }
}