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
求得节点数量。如果不是,则正常计算节点数目。继续利用后序遍历比较方便。