【树】leetcode_637_二叉树的层平均值

最新推荐文章于 2023-05-18 14:18:02 发布

原创最新推荐文章于 2023-05-18 14:18:02 发布 · 107 阅读

0 ·

CC 4.0 BY-SA版权

刷题的日常专栏收录该内容

54 篇文章

订阅专栏

该代码实现了一个C++函数，用于计算给定二叉树每一层节点值的平均数。通过广度优先搜索遍历树的每一层，计算并存储每层的平均值，最后返回结果向量。

【树】leetcode_637_二叉树的层平均值

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        TreeNode *cur = root;

        vector<double> result; 
        if (root == nullptr) return {};
        queue<TreeNode*> q_node;

        q_node.push(root);
        while(!q_node.empty()){
            int size = q_node.size();
            double re = 0;
            for(int i = 0; i<size; i++){
                cur = q_node.front();
                q_node.pop();
                re += cur->val;
                
                if (cur->left){
                    q_node.push(cur->left);
                }
                if (cur->right){
                    q_node.push(cur->right);
                }
            }
            cout<<re;
            result.push_back(re/double(size));
        }
        return result;
    }
};