이진 트리의 수준 평균

5089 단어 javascriptleetcode
이진 트리의 루트가 주어지면 각 수준의 노드 평균값을 배열 형태로 반환합니다. 실제 답변의 10-5 이내 답변이 허용됩니다.

예 1:
입력: 루트 = [3,9,20,null,null,15,7]
출력: [3.00000,14.50000,11.00000]
설명: 레벨 0의 노드 평균값은 3, 레벨 1은 14.5, 레벨 2는 11입니다.
따라서 [3, ​​14.5, 11]을 반환합니다.

/**
 * 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 === null) {
    return null;
  }
  let q = [root];
  let averageArray = [];

  while (q.length) {
    let level = [];
    let sum = 0;
    let count = 0;
    let loopTill = q.length;
    // looping only till the length of q;
    while (loopTill) {
      // removing first element from queue
      let node = q.shift();
      // adding up node val to get sum & also incrementing count tokeep track of nodes at each level
      sum = sum + node.val;
      count = count + 1;
      if (node.left !== null) {
        level.push(node.left);
      }
      if (node.right !== null) {
        level.push(node.right);
      }
      // reducing value as we have remove first element using shift() above
      loopTill--;
    }
    //get average of this level
    let averageOfLevel = sum / count;
    averageArray.push(averageOfLevel);

    q = level;
  }
  return averageArray;
};

좋은 웹페이지 즐겨찾기