剑指 Offer树的子结构

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。

思路

如果根节点相同则递归调用 isSubtreeWithRoot()，如果根节点不相同，则判断 root1 的左子树和 roo2 是否相同，再判断右子树和 root2 是否相同;
注意节点为空的条件，isSubStructure() 中，只要有树为空就返回 false; isSubtreeWithRoot() 中，要先判断 root2，如果 root2 为空，则说明第二棵树遍历完了，即匹配成功。

参考代码

class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (A == null || B == null) {
            return false;
        }

        return (isSubtreeWithRoot(A, B)  || isSubStructure(A.left, B) || isSubStructure(A.right, B));
    }

    public boolean isSubtreeWithRoot(TreeNode root1, TreeNode root2) {
            if (root2 == null) return true;
            if (root1 == null) return false;
            if (root1.val != root2.val) return false;
            return isSubtreeWithRoot(root1.left, root2.left) && isSubtreeWithRoot(root1.right, root2.right); 
    }
}