LeetCode 222. Count Complete Tree Nodes

LeetCode 222

最简单的方法就是遍历整个Tree，我们就知道数的节点的个数。
但是这样明显没有充分利用给的条件，并且需要遍历整个数，在complete tree的情况下不是最好的方法。我们知道如果是full complete tree，那么节点的个数是2的n次方减1 （n是高度）。利用这个条件，我们转换问题成，如果是full complete tree就利用这个方式做，如果不是那么就计算左子树和右子树的节点数。

    def countNodes(self, root: TreeNode) -> int:
        if root == None: return 0

        def levels(root:TreeNode):
            level = 0
            left, right = root, root
            while right:
                left = left.left
                right = right.right
                level +=1
            if left == None:
                return level, level
            return level +1, level

        leftLevels, rightLevels = levels(root)
        if leftLevels == rightLevels:
            return 2 ** leftLevels -1
        
        return 1 + self.countNodes(root.left) + self.countNodes(root.right)