【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
        

总结

相关推荐
BingLin-Liu7 分钟前
图论之cruskal算法(克鲁斯卡尔)
数据结构·算法·图论
进取星辰21 分钟前
PyTorch 深度学习实战(15):Twin Delayed DDPG (TD3) 算法
pytorch·深度学习·算法
strive-debug27 分钟前
C语言之 条件编译和预处理指令
算法
Sacuki1 小时前
BP神经网络公式推导与代码实现
算法
山登绝顶我为峰 3(^v^)31 小时前
VSCode + CMake
ide·vscode·算法·计算机·编辑器
勇敢滴勇2 小时前
【C++】二叉搜索树(二叉查找树、二叉排序树)详解
开发语言·c++·算法·霍夫曼树
f狐0狸x2 小时前
【蓝桥杯每日一题】3.16
c++·算法·蓝桥杯
周Echo周2 小时前
8、STL中的map和pair使用方法
开发语言·数据结构·c++·考研·算法·leetcode·pat考试
IT猿手3 小时前
基于牛优化( OX Optimizer,OX)算法的多个无人机协同路径规划(可以自定义无人机数量及起始点),MATLAB代码
人工智能·算法·机器学习·matlab·无人机
武乐乐~3 小时前
欢乐力扣:反转链表
算法·leetcode·链表