222. 完全二叉树的节点个数

最新推荐文章于 2024-06-14 09:50:01 发布

芒果香菠菠

最新推荐文章于 2024-06-14 09:50:01 发布

阅读量323

点赞数 9

文章标签： python 算法 leetcode

本文链接：https://blog.csdn.net/weixin_42396719/article/details/135540537

版权

文章介绍了如何使用递归和层序遍历算法计算二叉树的节点数量，同时提出了一种基于左右子树深度的优化方法。Solution类中的countNodes函数展示了这些计数策略。

摘要由CSDN通过智能技术生成

# 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:
    #     #递归
    #     return self.getNodeCount(root)

    # def getNodeCount(self, cur):
    #     if not cur:
    #         return 0
    #     return self.getNodeCount(cur.left) + self.getNodeCount(cur.right) + 1
        # #层序
        # if not root:
        #     return 0
        # queue = collections.deque([root])
        # nodecounts = 0
        # while queue:
        #     for _ in range(len(queue)):
        #         node = queue.popleft()
        #         nodecounts += 1
        #         if node.left:
        #             queue.append(node.left)
        #         if node.right:
        #             queue.append(node.right)
        # return nodecounts
        #左右深度比较
        if not root:
            return 0
        left = root.left
        right = root.right
        ldeep = 1
        rdeep = 1
        while left:
            ldeep += 1
            left = left.left
        while right:
            rdeep += 1
            right = right.right
        if ldeep == rdeep:
            return 2**ldeep - 1
        return self.countNodes(root.left) + self.countNodes(root.right) + 1