目录
- 1. 题目:
- 2. 我的代码:
- 小结:
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 = nodeleft_num = 0while left_node.left != None:left_num += 1left_node = left_node.left# 数当前节点的最深右节点的个数right_node = noderight_num = 0while right_node.right != None:right_num += 1right_node = right_node.rightif left_num == right_num:return 2 ** (left_num + 1) - 1else:return countN(node.left) + countN(node.right) + 1return countN(root)
统计完全二叉树的节点个数,可以使用普通方法(即随便一个遍历顺序求得节点数目)。但这里为了更加简便,利用完全二叉树的性质(即最后一层是从左到右满的)
对每一个节点进行判断,查看这个节点下面挂载的是否是满二叉树。如果是满二叉树,则利用公式2 ** 层数 - 1
求得节点数量。如果不是,则正常计算节点数目。继续利用后序遍历比较方便。
小结:
关注我给大家分享更多有趣的知识,以下是个人公众号,提供 ||代码兼职|| ||代码问题求解||
添加我的公众号即可: