给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 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) {