二叉树的层次遍历

题目:二叉树的层平均值

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.

示例 1:

输入:
3
/ \
9 20
/ \
15 7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/average-of-levels-in-binary-tree

解法：需要借助队列来进行遍历，将 节点的左右子树依次放入队列，然后再从队列中取节点，当队列为空时遍历结束。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def averageOfLevels(self, root: TreeNode) -> List[float]:
        if not root:
            return []
        res = [] #记录结果
        level = [] #每层的节点值列表
        stack = [root] #队列
        end =root #每层的边界节点（最右边节点）
        while stack:
            node = stack.pop(0) #取最左边的节点
            level.append(node.val)
            #如果该节点的左右子节点不为空时，依次加入到队列中
            if node.left:
                stack.append(node.left)
            if node.right:
                stack.append(node.right)
            #当节点是该层的最右边节点时，计算平均值
            if node == end:
                level_mean = sum(level)/len(level)
                res.append(level_mean)
                #对下一次循环的参数进行初始化               
                if stack:
                    end = stack[-1]#取下一层的最右边节点作为边界节点
                    level = []
        return res

来源：CSDN

作者：追梦杏花天影

链接：https://blog.csdn.net/u010569893/article/details/104678306