剑指 Offer 26. 树的子结构（python）

题目链接
题目描述：

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

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

例如:
给定的树 A:

在这里插入图片描述

给定的树 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

解题思路：

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        def judge(A,B):
            if not B:
                return True
            if not A:
                return False
            if A.val!=B.val:
                return False
            return judge(A.left,B.left) and judge(A.right,B.right)

        result=False
        if A and B:
            if A.val==B.val:
                result=judge(A,B)
            if not result:
                result=self.isSubStructure(A.left,B)
            if not result:
                result=self.isSubStructure(A.right,B)
        return result