leetcode 637 Average of Levels in Binary Tree

最新推荐文章于 2023-02-09 10:09:10 发布

MVincent

最新推荐文章于 2023-02-09 10:09:10 发布

阅读量224

点赞数

分类专栏： leetcode tree 文章标签： leetcode c python

本文链接：https://blog.csdn.net/MRxjh/article/details/83340906

版权

leetcode 同时被 2 个专栏收录

34 篇文章 0 订阅

订阅专栏

tree

4 篇文章 0 订阅

订阅专栏

python:

class Solution:
    def averageOfLevels(self, root):
        """
        :type root: TreeNode
        :rtype: List[float]
        """
        if root is None:
            return []
        res = []
        queue = [root]
        while(queue):
            sum_level = 0
            l = len(queue)
            for _ in range(l):
                node = queue.pop(0)
                sum_level += node.val
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(sum_level/l)
        return res

c++:

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        if (root == nullptr)
            return {};
        vector<double> res;
        queue<TreeNode*> q;
        q.push(root);
        
        while(!q.empty()){
            int l = q.size();
            double sum = 0.0;
            for (int i=0; i<l; i++){
                TreeNode* f = q.front();
                q.pop();
                sum += f->val;
                if(f->left) 
                    q.push(f->left);
                if(f->right)
                    q.push(f->right);
            }
            res.push_back(sum/l);
        }
        return res;
     }
};