给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.
示例 1:
输入: 3 / \ 9 20 / \ 15 7 输出: [3, 14.5, 11] 解释: 第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意:
- 节点值的范围在32位有符号整数范围内。
/**
* DEFINITION FOR A BINARY TREE NODE.
* STRUCT TREENODE {
* INT VAL;
* TREENODE *LEFT;
* TREENODE *RIGHT;
* TREENODE(INT X) : VAL(X), LEFT(NULL), RIGHT(NULL) {}
* };
*/
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
vector<double> ret;
queue<TreeNode*> que;
que.push(root);//从已有元素后面增加元素
while(!que.empty()) //查看是否为空范例,是的话返回1,不是返回0;
{
int size = que.size(); //输出现有元素的个数
double level = 0;
for(int i = 0; i < size; ++i)
{
TreeNode* node = que.front(); //显示第一个元素
que.pop(); //清除第一个元素
level += node->val;
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
}
ret.push_back(level/size);
}
return ret;
}
};
从题目的示例我们可以大概知道题目要求我们遍历二叉树的每层节点并计算其平均值。二叉树的搜索遍历无外乎这几种方法:先序遍历、中序遍历、后序遍历、BFS、DFS。结合题目,很显然BFS是比较适合题目的一种方法。
我们先往队列中push根节点,然后当队列不为空的时候,遍历整个队列。每遍历队列中的一个节点,就取出它的值并将其pop。此时顺带检查它的左右子节点是否为空,如果不为空,就将其子节点push进入队列中。这样,刚好遍历完一层节点,下一层的节点就全部存到队列中了。一层层循环下去直到最后一层,问题便顺利得到了解决。