【Leetcode】NO.637 二叉树的层平均值 (C++&Python) [二叉树]
2022/1/5 1:04:51
本文主要是介绍【Leetcode】NO.637 二叉树的层平均值 (C++&Python) [二叉树],对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
题目:637.二叉树的层平均值(Easy)
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组
思路
这题是在求解过二叉树的层序遍历之后,在来解决这道题就比较容易;
注意返回值是double;
代码
C++ 版本
/** * 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) { // 层序遍历,这个过程中出队列的时候保存最后一个元素 // 存放结果 vector<double> ans; // 存放double //存放节点 queue<TreeNode*> que; if(root) que.push(root); while(!que.empty()) { int size = que.size(); // 这个size很关键 double sum = 0,avg = 0; // 第一次提交失败是因为sum初始化为int for(int i=0;i<size;i++) { TreeNode* node = que.front(); que.pop(); sum += (node->val); if(node->left) que.push(node->left); if(node->right) que.push(node->right); } avg = sum/size; ans.push_back(avg); // 计算平均值 } for(int i=0;i<ans.size();i++) { cout<<ans[i]; } return ans; } };
Python 版本
这篇关于【Leetcode】NO.637 二叉树的层平均值 (C++&Python) [二叉树]的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-10-04Python外包有哪些常见的主要库-icode9专业技术文章分享
- 2024-10-02基于深度学习的钢铁缺陷检测系统(yolov8、注意力机制、PyQt5界面、Python)
- 2024-10-01怎么用python搭建一个系统-icode9专业技术文章分享
- 2024-09-30Python基础入门教程
- 2024-09-30获取参数学习:Python中的基础教程
- 2024-09-30Python编程基础详解
- 2024-09-29点击加载学习:Python编程基础教程
- 2024-09-29数据科学五大Python前端库:第二部分
- 2024-09-27使用python 将ETH账户的资产打散
- 2024-09-26Python编程基础