python leetcode 222. Count Complete Tree Nodes

Neekity

于 2018-12-17 14:05:49 发布

阅读量638

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode python 文章标签： python leetcode

本文链接：https://blog.csdn.net/Neekity/article/details/85048524

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种高效的算法，用于计算二叉树中节点的数量。通过比较左右子树的高度，该算法能够快速确定大部分节点的位置，从而避免了对整棵树的遍历。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution(object):
    def countNodes(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """ 
        if not root: return 0
        LH=self.getHeight(root.left)
        LR=self.getHeight(root.right)
        if LH==LR:
            return 2**LH+self.countNodes(root.right)
        else:
            return 2**LR+self.countNodes(root.left)
        
    def getHeight(self,root):
        if not root: return 0
        count=1
        while root.left:
            count+=1
            root=root.left 
        return count