树的子结构.py

'''
题目描述：
输入两棵二叉树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


解题思路：
直接递归 注意：子树为空的情况

'''


# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        #递归
        if not B:return False
        def recursive(roota,rootb):
            if not roota:return rootb is None
            if not rootb:return True
            if roota.val!=rootb.val:return False
            else:
                return recursive(roota.left,rootb.left) and recursive(roota.right,rootb.right)
        self.res=False
        def traverse(root):
            if  not root:return 
            res=recursive(root,B)
            print(res)
            if not self.res:
                self.res=res
            traverse(root.left)
            traverse(root.right)
            
            return 
        traverse(A)
        return self.res