1120 子树的最大平均值

1120 子树的最大平均值

题意

给你一棵二叉树的根节点 root，找出这棵树的 每一棵 子树的 平均值 中的 最大 值。
子树是树中的任意节点和它的所有后代构成的集合。
树的平均值是树中节点值的总和除以节点数。

示例

输入：[5,6,1]
输出：6.00000
解释：
以 value = 5 的节点作为子树的根节点，得到的平均值为 (5 + 6 + 1) / 3 = 4。
以 value = 6 的节点作为子树的根节点，得到的平均值为 6 / 1 = 6。
以 value = 1 的节点作为子树的根节点，得到的平均值为 1 / 1 = 1。
所以答案取最大值 6。

提示：
树中的节点数介于 1 到 5000之间。
每个节点的值介于 0 到 100000 之间。
如果结果与标准答案的误差不超过 10^-5，那么该结果将被视为正确答案。

题解

首先对于二叉树的问题大多可以归结为动态规划的问题。一般的思路是当要求的子问题涉及到是左子树还是右子树的时候，采用递归的解题方法，对两种问题的结果选择最优并且返回即可。
对于该问题，有两个方向，第一是需要求出以某一个节点的二叉树的节点总和，其次求出所包含的所有的节点。直接做除法即可。
第一步：因为需要计算平均值，所以需要用两个数字来保存当前的路径总和以及路径中的节点的总和

vector<int> arr{0,0};//其中0为所涉及的路径长度，1为节点的总数

第二步：
判断终止条件：当当前根节点为空的时候，则不存在二叉树，所以平均值为0

if (root == null) return 0;

第三步：
采用递归的思想，来计算左右子树的路径以及所包含的节点总数，然后加上当前根节点的数值即可计算平均值的最大值

double maxMean = 0;
 vector<int> dfs(TreeNode* root) {
        // arr[0] 保存当前节点所表示的子树元素之和，arr[1] 表示当前节点表示的子树的元素个数
        vector<int> arr{0,0};//初始化       
        if (root == null) return arr;
        vector<int> left = dfs(root->left);//由于dfs函数是计算以当前节点为根的最大的路径以及最大节点数，所以采用递归的方式，即可计算出左节点为根节点的二叉树的最大的路径以及最大的节点数
        vector<int> right = dfs(root->right);//由于dfs函数是计算以当前节点为根的最大的路径以及最大节点数，所以采用递归的方式，即可计算出右节点为根节点的二叉树的最大的路径以及最大的节点数
        //设置当前树的元素和
        //arr[0] = 左元素和 + 右元素和 + root->val;
        arr[0] = left[0] + right[0] + root->val;
        //设置节点个数
         //arr[0] = 左节点数 + 右节点数+当前节点;
        arr[1] = left[1] + right[1] + 1;
        //更新平均值
        maxMean = Math.max(maxMean, (double) arr[0] / arr[1]);//通过比较选择出最大的平均值，因为从该函数出发只是计算了一种结果的最大值
        return arr;
 }

总结：
对于二叉树的问题，大多数可以总结为以根节点为开始，采用递归的思想对左右子树进行计算。对于计算的结果通过与全局变量进行对比得出最优解