LeetCode 637. 二叉树的层平均值（层次遍历queue）

最新推荐文章于 2022-01-14 14:58:06 发布

Michael阿明

最新推荐文章于 2022-01-14 14:58:06 发布

阅读量731

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_21201267/article/details/100717726

版权

LeetCode 专栏收录该内容

1849 篇文章 126 订阅

订阅专栏

1. 题目

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
在这里插入图片描述

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

2. 解题

queue按层遍历，每次进入循环记录队列长度n

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        if(root == NULL)
        	return {};
        vector<double> ans;
        int Num, n;
        long lvsum;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
        	lvsum = 0;
        	n = Num = q.size();
        	while(Num--)
        	{
        		lvsum += q.front()->val;
        		if(q.front()->left)
        			q.push(q.front()->left);
                if(q.front()->right)
                    q.push(q.front()->right);
        		q.pop();
        	}
        	ans.push_back(double(lvsum)/n);
        }
        return ans;
    }
};