leetCode题解之求二叉树每层的平均值

1、题目描述

Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.

计算二叉树每一层的节点的数据域的平均值。

2、题目分析

使用广度优先遍历方法，逐层对二叉树进行访问，求均值。使用一个队列数据结构完成对二叉树的访问，队列（queue）的特点是FIFO，即先进先出，queue的几种常用的方法是:

queue::front()  :访问队首元素

queue:: pop()  删除队首元素

queue::push()  入队

queue:: back()  访问队尾元素

vector<double> averageOfLevels(TreeNode* root) {
        
        vector<double> ave;
        queue<TreeNode*> q;
        q.push(root);

        while(!q.empty())
            {
                double temp = 0.0;
                int s = q.size();
                for( int i = 0; i < s;i++ )
                    {
                        temp += q.front()->val;
                        if(q.front()->left) q.push(q.front()->left);
                        if(q.front()->right ) q.push(q.front()->right);
                        q.pop();
                    }
                ave.push_back(temp/s);

            }
        return ave;
        
        
        
    }

3、代码