本文主要是介绍[Tree Breadth First Search] 二叉树的层平均值,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
leetcode 637,Average of Levels in Binary Tree,难度,easy
0. 题干
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
示例 1:
输入:
输出:[3, 14.5, 11]
解释:
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。
提示:
节点值的范围在32位有符号整数范围内。
1. 代码(含注释)
解题思路
对二叉树进行广度优先遍历,遍历到每一层时,计算均值,将结果添加到数组中。
思路还是BFS,代码没啥说的,比较简单。
class Solution {
public:vector<double> averageOfLevels(TreeNode* root) {queue<TreeNode*> que; //构造队列que,里面的元素都是TreeNode*类型的指针que.push(root);//广度优先遍历,先将根节点添加到队列vector<double> res; //存储结果while (que.size()!= 0) {int n = que.size();double sum=0; //sum用来存储每一层节点值的和for ( int i=0;i<n;i++){TreeNode* tmp=que.front(); //取出队列的第一个元素que.pop();sum=sum+tmp->val; //存储每一层的节点和//如果该节点的左右孩子存在,就放到队列里面去if(tmp->left)que.push(tmp->left);if(tmp->right)que.push(tmp->right);}res.push_back(sum/n);}return res;}
};
这篇关于[Tree Breadth First Search] 二叉树的层平均值的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!