Skip to main content

二叉树的层平均值

题目​

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

提示:
  • -2³¹ <= Node.val <= 2³¹ - 1
示例
  3
 / \
9 20
  / \
 15  7


输入: root = [3,9,20,null,null,15,7]
输出: [3.00000,14.50000,11.00000]
解释: 第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11
因此返回 [3, 14.5, 11]
    3
   / \
  9  20
 / \
15  7

输入: root = [3,9,20,15,7]
输出: [3.00000,14.50000,11.00000]

题解​

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var averageOfLevels = function (root) {
  if (!root) return []
  const result = []
  const queue = [root]

  while (queue.length > 0) {
    const n = queue.length

    let ans = 0
    let count = 0
    for (let i = 0; i < n; i++) {
      const node = queue.shift()
      ans += node.val
      count++

      if (node.left) queue.push(node.left)
      if (node.right) queue.push(node.right)
    }

    result.push(ans / count)
  }

  return result
}