代码随想录 Leetcode637. 二叉树的层平均值

题目:

代码随想录 Leetcode637. 二叉树的层平均值_第1张图片


代码(首刷自解 2024年1月24日):

class Solution {
public:
    vector averageOfLevels(TreeNode* root) {
        vector res = {};
        if(root == nullptr) return res;
        queue que;
        TreeNode* cur = root;
        que.push(cur);
        int size = 0;
        while (!que.empty()) {
            size = que.size();
            double digit = size;
            long long sum = 0;
            while (size--) {
                TreeNode* node = que.front();
                sum += node->val;
                que.pop();
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            res.emplace_back(sum/digit);
        }
        return res;
    }
};

你可能感兴趣的:(#,leetcode,---,easy,算法,leetcode,c++)