【面试经典 150 | 二叉树层序遍历】二叉树的层平均值

文章目录

写在前面
Tag
题目来源
题目解读
- 方法一：层序遍历
写在最后

写在前面

本专栏专注于分析与讲解【面试经典150】算法，两到三天更新一篇文章，欢迎催更……

专栏内容以分析题目为主，并附带一些对于本题涉及到的数据结构等内容进行回顾与总结，文章结构大致如下，部分内容会有增删：

Tag：介绍本题牵涉到的知识点、数据结构；
题目来源：贴上题目的链接，方便大家查找题目并完成练习；
题目解读：复述题目（确保自己真的理解题目意思），并强调一些题目重点信息；
解题思路：介绍一些解题思路，每种解题思路包括思路讲解、实现代码以及复杂度分析；
知识回忆：针对今天介绍的题目中的重点内容、数据结构进行回顾总结。

Tag

【二叉树遍历】【层序遍历】

题目来源

637. 二叉树的层平均值

题目解读

方法一：层序遍历

本题比较基础，在二叉树层序遍历的基础上，加上枚举当前层所有节点并将节点值加起来求平均值即可。对二叉树层序遍历不熟悉的同学可以参考广度优先搜索（1）之树的层序遍历。

代码

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:vector<double> averageOfLevels(TreeNode* root) {queue<TreeNode*> que;vector<double> res;que.push(root);TreeNode* curr;while (!que.empty()) {int size = que.size();double sum = 0.0;for (int i = 0; i < size; ++i) {curr = que.front(); que.pop();sum += curr->val;if (curr->left) {que.push(curr->left);}if (curr->right) {que.push(curr->right);}}res.push_back(sum / size);}return res;}
};

复杂度分析

时间复杂度： $O (n)$ ， $n$ 是二叉树的节点个数。

空间复杂度： $O (n)$ 。