二分查找-6.7 完全二叉树计数-CSDN博客

本文链接：https://blog.csdn.net/Iamlzm/article/details/130061845

题目：

给定一棵完全二叉树的根节点root，返回这棵树的节点个数。如果完全二叉树的节点数为N，请实现时间复杂度低于O(N)的解法。

给定树的根结点root，请返回树的大小。

测试用例：

输入：{1,2,3}

输出：3

思路：

完全二叉树的基本概念不做赘述。计算一棵树上的结点数最简单的方法当然是做遍历。但是这里要求用时间复杂度小于O(n)的方法来计数。

假设我们知道了一棵子树是满二叉树，以及它的高度，那么这颗子树的结点数计算起来就很容易。根据完全二叉树的定义，最后一层（记为第h层）的结点数取值可能为[1: $2^{h-1}$ ]，假如h层某一个结点不为空，那么在它左边的兄弟节点都不为空；假如h层某一个结点为空，那么在它右边的兄弟节点都为空。所以，假设根节点的右子树上最左边的结点不为空，说明根节点的左子树是一颗满二叉树；而如果根节点的右子树上最左边的结点为空，说明根结点的右子树也是一颗满二叉树，只是高度比左子树低一层。下面两幅图即是这两种情况的说明。

所以，对于一颗高度为h的完全二叉树，假如右子树高度为h-1，那么左子树+根节点的数目 = 1 + 2 + $2^{h-2}$ + 1 = $2^{h-1}$ ，然后可以递归地查找右子树；假如右子树高度为h-2，那么右子树+根节点数目 = 1 + 2 + $2^{h-3}$ + 1 = $2^{h-2}$ ，然后递归地查找左子树。边界条件即查找到第h层时，此时说明只剩这一个叶子节点了，返回1即可

代码：

# -*- coding:utf-8 -*-

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class CountNodes:
    def count(self, root):
        # write code here
        if not root:
            return 0
        return self.getNum(root, 1, self.getHeight(root, 1))
    
    def getNum(self, root, level, height):
        if level == height:
            return 1
        if self.getHeight(root.right, level+1) == height:
            return (1 << (height-level)) + self.getNum(root.right, level+1, height)
        else:
            return (1 << (height-level-1)) + self.getNum(root.left, level+1, height)

    def getHeight(self, root, level):
        while root:
            root = root.left
            level += 1
        return level-1