lintcode 具有最大平均数的子树

最新推荐文章于 2025-06-26 16:04:08 发布

锤某

最新推荐文章于 2025-06-26 16:04:08 发布

阅读量514

点赞数

CC 4.0 BY-SA版权

分类专栏：每日一练 C++

本文链接：https://blog.youkuaiyun.com/qq_40147449/article/details/88676558

每日一练同时被 2 个专栏收录

58 篇文章

订阅专栏

C++

12 篇文章

订阅专栏

博客围绕在二叉树中寻找有最大平均值的子树展开。给出了具体样例，如输入特定结构的二叉树，输出最大平均值子树的根结点。还阐述了思考过程，即使用函数按后根遍历顺序遍历节点，用左右子树和计算根的和并保存最大平均数。

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

描述

给一棵二叉树，找到有最大平均值的子树。返回子树的根结点

样例

样例 1

输入：
1
/
-5 11
/ \ /
1 2 4 -2
输出：11(它是一个TreeNode)
样例 2

输入：
1
/
-5 11
输出：11(它是一个TreeNode)

思考

使用一个函数来遍历这个二叉树。
根据后根遍历的顺序遍历每个节点，用左右子树的和来计算根的和，并且一直保存一个最大的平均数。如果左右子树没有节点，那么根直接返回自己的值，和大小为1。

代码

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: the root of binary tree
     * @return: the root of the maximum average of subtree
     */
    TreeNode * findSubtree2(TreeNode * root) {
        // write your code here
        if (!root || (!root->left && !root->right))
            return root;
        double maxAvg = INT_MIN;
        TreeNode *maxNode = NULL;
        int size = 0;
        recur(root,maxNode, size, maxAvg);
        return maxNode;
    }
    
    double recur(TreeNode *node, TreeNode * &maxNode, int &size, double& maxAvg) {
        if (!node) return 0;
        int leftSize = 0, rightSize = 0;
        double leftAvg,rightAvg;
        leftAvg = recur(node->left, maxNode, leftSize, maxAvg);
        rightAvg = recur(node->right, maxNode, rightSize, maxAvg);
        size = leftSize+rightSize+1;
        double Avg = (leftAvg * leftSize + rightAvg * rightSize + node->val)/(double)size;
        if (Avg > maxAvg) {
            maxAvg = Avg;
            maxNode = node;
        }
        return Avg;
        
    }
};