leetcode代码记录(完全二叉树的节点个数

简介: leetcode代码记录(完全二叉树的节点个数

1. 题目:

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

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

示例 1:

输入:root = [1,2,3,4,5,6]

输出:6

示例 2:

输入:root = []

输出:0

示例 3:

输入:root = [1]

输出:1

2. 我的代码:

# 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:
        # 后序遍历
        # 递归
        def countN(node):
            # 终止条件
            if node == None:
                return 0

            # 数当前节点的最深左节点的个数
            left_node = node
            left_num = 0
            while left_node.left != None:
                left_num += 1
                left_node = left_node.left
            # 数当前节点的最深右节点的个数
            right_node = node
            right_num = 0
            while right_node.right != None:
                right_num += 1
                right_node = right_node.right

            if left_num == right_num:
                return 2 ** (left_num + 1) - 1
            else:
                return countN(node.left) + countN(node.right) + 1
            
        return countN(root)

统计完全二叉树的节点个数,可以使用普通方法(即随便一个遍历顺序求得节点数目)。但这里为了更加简便,利用完全二叉树的性质(即最后一层是从左到右满的)

对每一个节点进行判断,查看这个节点下面挂载的是否是满二叉树。如果是满二叉树,则利用公式2 ** 层数 - 1求得节点数量。如果不是,则正常计算节点数目。继续利用后序遍历比较方便。

目录
相关文章
|
5天前
|
机器学习/深度学习
leetcode代码记录(旋转图像
leetcode代码记录(旋转图像
10 0
|
5天前
|
算法
leetcode代码记录(全排列 II
leetcode代码记录(全排列 II
13 4
|
5天前
|
算法
leetcode代码记录(全排列
leetcode代码记录(全排列
12 1
|
5天前
|
索引
leetcode代码记录(Z 字形变换
leetcode代码记录(Z 字形变换
11 1
|
5天前
leetcode代码记录(最长回文子串
leetcode代码记录(最长回文子串
10 2
|
5天前
leetcode代码记录(回文数
leetcode代码记录(回文数
13 1
|
5天前
|
算法
leetcode代码记录(寻找两个正序数组的中位数
leetcode代码记录(寻找两个正序数组的中位数
13 2
|
5天前
leetcode代码记录(两数之和
leetcode代码记录(两数之和
11 1
|
5天前
|
索引
leetcode代码记录(最长公共子序列
leetcode代码记录(最长公共子序列
7 0
|
5天前
|
索引
leetcode代码记录(最长重复子数组
leetcode代码记录(最长重复子数组
12 0