用 Python 编写程序来查找二叉树的叶子节点和非叶子节点

假设我们有一棵二叉树，我们必须找到一个由两个数字组成的列表，其中第一个数字是树中叶子的数量，第二个数字是非叶子节点的数量。

因此，如果输入如下

那么输出将是 (3, 2)，因为有 3 个叶子节点和 2 个非叶子节点。

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

• 如果 n 为空，则
  • 返回 (0, 0)
• 如果 n 的左侧为空且 n 的右侧为空，则
  • 返回 (1, 0)
• left := resolve(left of n)
• right := resolve(right of n)
• 返回 (left[0] + right[0], 1 + left[1] + right[1])

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

示例

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.val = data
      self.left = left
      self.right = right
class Solution:
   def solve(self, n):
      if not n:
         return 0, 0
      if not n.left and not n.right:
         return 1, 0
      left, right = self.solve(n.left), self.solve(n.right)
      return left[0] + right[0], 1 + left[1] + right[1]
ob = Solution()
root = TreeNode(6)
root.left = TreeNode(2)
root.right = TreeNode(6)
root.right.left = TreeNode(10)
root.right.right = TreeNode(2)
print(ob.solve(root))

输入

root = TreeNode(6)
root.left = TreeNode(2)
root.right = TreeNode(6)
root.right.left = TreeNode(10)
root.right.right = TreeNode(2)

输出

(3, 2)

相关文章

• Python 中元组列表到字典的转换
• Python - 从两个列表中配对，使得成对的元素不相同
• 使用 Python 和 SQLite 的 SQL
• 使用 pytrie 模块在 Python 中进行前缀匹配
• 创建和访问 Python 包
• 如何使用正则表达式和数据类型选择多个 DataFrame 列
• 如何在 Pandas 中查找和过滤重复行?
• 用 Python 编写程序以正确的顺序查找机场?
• 如何在 Python Pandas 中使用字典顺序切片选择数据子集?
• 用 Python 编写程序检查我们是否可以从给定的字符串字符中创建 k 个回文?

有用资源

• python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/