222完全二叉树的节点个数

# 执行用时 :124 ms, 在所有 python3 提交中击败了29.46% 的用户
# 内存消耗 :20 MB, 在所有 python3 提交中击败了98.68%的用户
class Solution:
    def countNodes(self, root: TreeNode) -> int:
        stack = []
        stack_next = []
        level = 0
        if root is None:
            return 0
        stack_next.append(root)
        while stack_next:
            stack = stack_next
            stack_next = []
            l = len(stack)
            # 非空的list才能pop()
            while stack:
                # 注意这里用的是队列不是栈,为的是保证每次都是从左往右遍历
                a = stack.pop(0)
                if a:
                    stack_next.append(a.right)
                    stack_next.append(a.left)
                # 一旦遇到某个节点为空了,说明到最后一行了,直接计算并返回结果
                else:
                    return int(len(stack_next)/2) + (2 ** (level-1)) - 1
            level += 1
        # 原先的代码比较耗时,最后一行可能会无效遍历,
        # return (2 ** (level-2) - 1 + int(l/2))