用 Python 编写程序检查所有叶子是否处于同一级别

pythonserver side programmingprogramming更新于 2026/1/19 7:08:17

假设我们有一棵二叉树；我们必须检查所有叶子是否处于同一级别。

因此，如果输入如下

则输出将为 True

为了解决这个问题，我们将遵循以下步骤 −

定义一个函数 dfs() 。这将获取 root，d
如果 root 不为空，则
- 如果 root 的左侧为空且 root 的右侧为空，则
  - 在深度末尾插入 d
- 否则，
  - dfs(root 的左侧，d + 1)
  - dfs(root 的右侧，d + 1)
从主方法中，执行以下操作 −
depth := a new list
dfs(root， 0)
当深度只有一个值时返回 true

让我们看看下面的实现以便更好地理解 −

示例

class TreeNode:
   def __init__(self, value):
      self.val = value
      self.left = None
      self.right = None

class Solution:
   def solve(self, root):
      self.depth = []
      self.dfs(root, 0)
      return len(set(self.depth)) == 1
   def dfs(self, root, depth):
      if root:
         if not root.left and not root.right:
            self.depth.append(depth)
         else:
            self.dfs(root.left, depth + 1)
            self.dfs(root.right, depth + 1)
ob = Solution()
root = TreeNode(5)
root.left = TreeNode(4)
root.left.left = TreeNode(2)
root.right = TreeNode(10)
root.right.left = TreeNode(7)
root.right.right = TreeNode(15)
print(ob.solve(root))

输入

root = TreeNode(5)
root.left = TreeNode(4)
root.left.left = TreeNode(2)
root.right = TreeNode(10)
root.right.left = TreeNode(7)
root.right.right = TreeNode(15)