LeetCode刷题（147）～二叉树的层平均值

题目描述

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

示例 1：

输入： 3 / \
  9  20 /  \ 15   7
输出：[3, 14.5, 11]
解释：
第 0 层的平均值是 3 ,  第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
 

提示：

• 节点值的范围在32位有符号整数范围内。

解答 By 海轰

提交代码

vector<double> averageOfLevels(TreeNode* root) { queue<TreeNode*> q; vector<double> ans; q.push(root); while(!q.empty()) { int queuesize=q.size(); double sum=0; for(int i=0;i<queuesize;++i) { TreeNode* temp=q.front(); q.pop(); sum+=(temp->val); if(temp->left) q.push(temp->left); if(temp->right) q.push(temp->right); } ans.push_back(sum/queuesize); } return ans; }

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
 

运行结果

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/average-of-levels-in-binary-tree

文章来源: haihong.blog.csdn.net，作者：海轰Pro，版权归原作者所有，如需转载，请联系作者。

原文链接：haihong.blog.csdn.net/article/details/108545149