目录

一、题目内容

二、解题思路

三、代码

一、题目内容

给出一个完全二叉树，求出该树的节点个数。

说明：

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

示例:

输入: 
    1
   / \
  2   3
 / \  /
4  5 6

输出: 6

二、解题思路

1.先左后右访问得到子树高度；

2.相等则说明左子树是满二叉树，判断右子树节点个数；

3.不相等则说明右子树高度小于左子树，则判断左子树高度即可；

例：

4+1+1（1->2->4->5,3,6）

三、代码

      # Definition for a binary tree node.
      class TreeNode:
      def __init__(self, x):
       self.val = x
       self.left = None
       self.right = None
      def __repr__(self):
      return str(self.val)
      class Solution:
      def countNodes(self, root: TreeNode) -> int:
      if root is None:
      return 0
       left_h = self.get_height(root.left)
       right_h = self.get_height(root.right)
      if left_h == right_h:
      return self.countNodes(root.right) + (1 << left_h)
      else:
      return self.countNodes(root.left) + (1 << right_h)
      def get_height(self, root):
      if root is None:
      return 0
       h = 0
      while root is not None:
       h += 1
       root = root.left
      return h
      if __name__ == '__main__':
       a = TreeNode(1)
       a.left = TreeNode(2)
       a.right = TreeNode(3)
       a.left.left = TreeNode(4)
       a.left.right = TreeNode(5)
       a.right.left = TreeNode(6)
      # a.right.right = TreeNode(7)
       s = Solution()
       ans = s.countNodes(a)
       print(ans)
  
 

文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。

原文链接：nickhuang1996.blog.csdn.net/article/details/110174149