给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
输入:
3
/ \
9 20
/ \
15 7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意:
- 节点值的范围在32位有符号整数范围内。
思路:
既然是求每一层的平均值,那自然是要用到二叉树的层序遍历了。但不全是层序遍历。
当队列不为空时,队列此时的长度代表着这一层的结点数,每次不是取一个节点,而是取一层的节点,这样才可以算出这一层的平均值。
JavaScript的数组既可以做栈也可以做队列,简直不要太方便!
JavaScript中与队列出元素对应的操作是 shift(),删除第一个元素并返回这个元素。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number[]}
*/
var averageOfLevels = function(root) {
var numbers = [];
var queue = []; // 层序遍历的队列
if (root !== null) {
queue.push(root);
}
while (queue.length !== 0) {
var sum = 0;
var num = queue.length; // l代表此时当前层的节点数
for (var i = 0; i < num; i ++) {
var currentNode = queue.shift(); // shift:删除数组的第一个元素并返回
sum += currentNode.val;
if (currentNode.left !== null) {
queue.push(currentNode.left);
}
if (currentNode.right !== null) {
queue.push(currentNode.right);
}
}
numbers.push(sum / num); // 将当前层的平均数添加到数组
}
return numbers;
};