《剑指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
遍历父树的每个节点,然后和子树做对比
时间复杂度 O(MN)
空间复杂度 O(M)


class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        def equal(org, target):
            if not target:
                return True
            if not org or org.val != target.val:
                return False
            return equal(org.left, target.left) and equal(org.right, target.right)
        def search_root(father, son):
            return bool(father and son) and (equal(father, son) or search_root(father.left, son) or search_root(father.right, son))
        return search_root(A, B)

 

你可能感兴趣的:(LeetCode点滴)