给你一棵二叉树，请你用分解问题的思路写一个 count 函数，计算这棵二叉树共有多少个节点

AI研习室

于 2025-01-11 17:18:29 发布

阅读量80

点赞数 2

文章标签：算法

本文链接：https://blog.youkuaiyun.com/xiangfengweixiaojie/article/details/145079872

版权

# 定义：输入一棵二叉树，返回这棵二叉树的节点总数
def count(root):
    if root is None:
        return 0
    # 当前节点关心的是两个子树的节点总数分别是多少
    # 因为用子问题的结果可以推导出原问题的结果
    leftCount = count(root.left)
    rightCount = count(root.right)
    # 后序位置，左右子树节点数加上自己就是整棵树的节点数
    return leftCount + rightCount + 1