力扣题目——637. 二叉树的层平均值

注：本文的代码实现使用的是 JS（JavaScript），为前端中想使用JS练习算法和数据结构的小伙伴提供解题思路。

描述

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

---

示例:

输入：

    3
   / \
  9  20
    /  \
   15   7

输出：[3, 14.5, 11]
解释：第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

---

解题思路

这道题只需要在层序遍历法的结果的基础上，将每一层节点累计求和后除以这一层的节点数即可。

点击链接即可查看二叉树层序遍历算法的详解 力扣题目——102. 二叉树的层序遍历

层序遍历代码

var levelOrder = function(root) {
	// 用于存放最后结果的数组
    let result = []
    // 初始化队列，若 root 为 null，则队列直接为[]， 不进行 while 循环，直接返回result
    let queue = root ? [root] : []
    // 只要队列中存在元素，则遍历，若不存在元素，说明树已经遍历完毕，退出
    while(queue.length !== 0){
    	// 存放每一层元素
        let temp = []
		// 遍历 len 个元素，即遍历这一层的所有元素
        for(let i = 0, len = queue.length; i < len; i++){
            let node = queue.shift()
            temp.push(node.val)
            // 只有不为空的时候，才需要将子节点 push 到队列中
            if(node.left) queue.push(node.left)
            if(node.right) queue.push(node.right)
        }
        result.push(temp)
    }
    return result
};

这道题目的代码

var averageOfLevels = function(root) {
    let queue = root === null ? [] : [root]
    let result = []
    while(queue.length !== 0){
        let average = 0
        let count = 0
        for(let i = 0, len = queue.length; i < len; i++){
            const node = queue.shift()
            // 对某一层的节点值求和
            average += node.val
            // 统计某一层的节点个数
            count++
            if(node.left) queue.push(node.left)
            if(node.right) queue.push(node.right)
        }
        // 保证 count 不为0
        if(count) average /= count
        result.push(average)
    }
    return result
};