637. 二叉树的层平均值 C++

最新推荐文章于 2021-08-30 20:14:27 发布

ZJU宁波理工

最新推荐文章于 2021-08-30 20:14:27 发布

阅读量319

点赞数

文章标签： leetcode C++

本文链接：https://blog.csdn.net/weixin_40659011/article/details/102250906

版权

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

在这里插入图片描述

简单题层次遍历

vector<double> averageOfLevels(TreeNode* root) 
{
	vector<double>res;
	if (root == NULL)
		return res;
	else
	{
		queue<TreeNode *>temp;
		temp.push(root);
		while (temp.empty())
		{
			int s = temp.size();
			double count = 0;
			for (int i = 0; i < s; i++)
			{
				TreeNode* p = temp.front();
				count += p->val;
				if (p->left)
					temp.push(p->left);
				if (p->right)
					temp.push(p->right);
				temp.pop();
			}
			res.push_back(count*1.0 / s);
		}
		return res;
	}
}