TIME:
凌晨🕛

描述

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

思路

BFS 一层一层遍历
需要注意左右节点是否null

标签

队列 shift push

代码

var averageOfLevels = function (root) {
    let resArray = [];
    let levelQueue = [root]

    while (levelQueue.length) {
        const size = levelQueue.length
        let = tempSum = 0;
        for (let i = 0; i < size; i++) {
            const cur = levelQueue.shift()

            tempSum = tempSum + cur?.val
            if (cur?.left) {
                levelQueue.push(cur?.left)
            }
            if (cur?.right) {
                levelQueue.push(cur?.right)

            }
        }
        resArray.push(tempSum / size)
    }
    return resArray
};