zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【Leetcode刷题Python】剑指 Offer 26. 树的子结构

PythonLeetCode Offer 刷题 26
2023-09-14 09:13:02 时间

1 题目

输入两棵二叉树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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2 解析

(1)首先如果两个子树都为空,不成立
(2)判断A的左子树和B是否为包含关系,或者A的右子树是否为包含关系,或者递归去找所有子树是否包含了,只要有一个返回True即可。

3 python实现

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