Leetcode 6057 统计值等于子树平均值的节点数

Llocyt

于 2022-05-08 22:56:10 发布

阅读量266

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode 深度优先算法

本文链接：https://blog.youkuaiyun.com/Llocyt/article/details/124656566

这篇博客介绍了一种解决二叉树问题的方法，通过深度优先搜索（DFS）检查每个节点是否为子树的平均值，并利用全局变量记录状态。代码中定义了两个全局变量`temp`和`ans`，分别用于累计节点值和计数，当子树的平均值等于根节点值时答案加一。这种方法虽然简单易懂，但全局变量的使用可能引入一些复杂性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这题是今天第292场周赛的第二题

我就用一套小白解法来解题吧

我想的是用dfs，再写一个函数来检查一下是否为子树的平均值，如果是的话，就答案加一

上代码

class Solution {
public:
    int ans = 0;
    int sum = 0;
    int temp = 0;
    void checktree(TreeNode* root)
    {
        if(root == NULL) return;
        sum = 0;
        temp = 0;
        dfs(root);
        if (sum / temp == root->val)
        {
            ans ++;
        }
        checktree(root->left);
        checktree(root->right);
    }
    void dfs(TreeNode* root)
    {   
        if(root == NULL) return;
        if(root != nullptr)
        {
            sum += root->val;
            temp ++;
            dfs(root->left);
            dfs(root->right);
        }
    }
    int averageOfSubtree(TreeNode* root) 
    {
        checktree(root);
        return ans;
    }
};

用的全局变量temp，ans和sum

如果定义在函数里的话，很显然会报错的，我们check函数嵌套dfs，即可检测出答案，简单易懂。