路飞一起刷 LeetCode 剑指 Offer 26. 树的子结构

前端

2023-10-26 20:52:52

大家好，我是挨打的阿木木，一位热衷于算法的前端摸鱼老。我打算定期与大家分享我在刷算法题时的一些思考和心得。如果你也是一位渴望提升自我逼格的摸鱼老，欢迎关注我，一起踏上学习的征程。

今天，我们就来共同探究《剑指 Offer》中的第 26 题：“树的子结构”。

题目

给定两棵二叉树 A 和 B，判断 B 是否是 A 的子结构。

也就是说，如果我们把 B 作为一个节点插入到 A 中，B 的所有节点都能成为 A 的子节点，那么我们就可以说 B 是 A 的子结构。

举例说明

输入：
- A = [1, 2, 3, 4, 5]
- B = [2, 4]
输出：true
解释：B 是 A 的子结构，因为我们可以将 B 插入到 A 中得到新的二叉树 [1, 2, 3, 4, 5, 2, 4]，其中 B 的所有节点都成为了 A 的子节点。
输入：
- A = [1, null, 2, 3, 4, 5]
- B = [2, 4]
输出：false
解释：B 不是 A 的子结构，因为如果我们将 B 插入到 A 中，B 的节点 2 的右子节点 4 无法成为 A 的子节点。

解题思路

为了判断 B 是否是 A 的子结构，我们可以采用递归的方法：

首先，我们需要比较 A 和 B 的根节点。如果它们相等，则进一步检查 B 的左子树和右子树是否分别等于 A 的左子树和右子树。
如果 A 和 B 的根节点不相等，则我们继续检查 A 的左子树和右子树是否包含 B 的子结构。
如果在任何情况下，我们发现 B 的所有节点都能够成为 A 的子节点，则说明 B 是 A 的子结构。否则，B 不是 A 的子结构。

代码实现

def is_subtree(A, B):
  if not A:
    return False
  if not B:
    return True
  
  if A.val == B.val:
    return is_subtree(A.left, B.left) and is_subtree(A.right, B.right)
  else:
    return is_subtree(A.left, B) or is_subtree(A.right, B)