【LeetCode222】完全二叉树的节点个数

题目描述

给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。

完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层(从第 0 层开始),则该层包含 1~ 2h 个节点。

思路与算法

  1. base cases:
    • 空树:节点的数量是 0
    • 完美二叉树:2^d−1
  2. 递归步骤
    • 递归计算左子树中的节点数
    • 递归计算右子树中的节点数
    • 为当前根节点加 1
  3. 结果组合

代码

python3 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0

        left_depth = self.getLeftDepth(root)
        right_depth = self.getRightDepth(root)
        if left_depth == right_depth:
            return (1 << left_depth) - 1 # same as 2^left_depth - 1

        return 1 + self.countNodes(root.left) + self.countNodes(root.right)

    def getLeftDepth(self, node: TreeNode) -> int:
        depth = 0
        while node:
            depth += 1
            node = node.left
        return depth
    
    def getRightDepth(self, node: TreeNode) -> int:
        depth = 0
        while node:
            depth += 1
            node = node.right
        return depth
        

总结

相关推荐
Dream it possible!1 小时前
CCF CSP 第30次(2023.09)(1_坐标变换_C++)(先输入再计算;边输入边计算)
c++·算法·csp
Terrence Shen1 小时前
跟着AI复习一下pytorch原理和操作
人工智能·pytorch·python·深度学习·算法·机器学习
军训猫猫头2 小时前
80.Dictionary 字典 C#例子
开发语言·算法·c#
码农幻想梦2 小时前
实验5 逻辑回归
算法·机器学习·逻辑回归
听微雨2 小时前
P9241 [蓝桥杯 2023 省 B] 飞机降落
算法·深度优先
平生不喜凡桃李3 小时前
对最近的刷题做一个小总结(关于动态规划和贪心)
算法·动态规划·双指针
zhangzeyuaaa3 小时前
深度学习有哪些算法?
人工智能·深度学习·算法
CodeJourney.3 小时前
探索ima.copilot:个人知识库搭建的AI新利器
数据库·人工智能·python·算法
天行者@4 小时前
OTSU算法(大津算法)
人工智能·opencv·算法·二值化
小码农<^_^>4 小时前
数据结构--算法篇
数据结构·算法