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
};
