LeetCode 637. 二叉树的层平均值（深度优先搜索dfs）

岁忧

于 2024-08-26 10:59:05 发布

阅读量79

点赞数 2

分类专栏： LeetCode 文章标签：深度优先 leetcode 算法 c++ 数据结构

本文链接：https://blog.csdn.net/weixin_46028214/article/details/141558582

版权

LeetCode 专栏收录该内容

424 篇文章 0 订阅

订阅专栏

题目：637. 二叉树的层平均值

在这里插入图片描述

思路：用数组v记录每层节点值的总和，数组ct记录每层节点的个数。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {

public:
    
    void dfs(TreeNode* root,int id,vector<double>& v,vector<int> &ct){
        if(root==nullptr) return;
        v[id]+=root->val;
        ct[id]++;
        dfs(root->left,id+1,v,ct);
        dfs(root->right,id+1,v,ct);
    }
    vector<double> averageOfLevels(TreeNode* root) {
    	//记录每层节点值的总和
        vector<double> v(10010,0);
        //记录每层节点的个数
        vector<int> ct(10010,0);
        dfs(root,0,v,ct);
        vector<double> avg;
        for(int i=0;i<10010;i++){
            if(ct[i]){
                avg.push_back(v[i]/ct[i]);
            }else{
                break;
            }
        }
        return avg;
    }
};