6057. 统计值等于子树平均值的节点数、树的后序遍历

思路：

 分析题意可知，可以借助树的后序遍历，来完成计算。这里我使用递归的方式编写后序遍历。

/**
 * 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:
    int averageOfSubtree(TreeNode* root) {
        // 保存最后结果
        int ans = 0;

        // 记录左右子树的情况
        int count0 = 0, val0 = 0;
        int count1 = 0, val1 = 0;

        // 树的后续遍历
        // 先遍历左子树
        recuriveCal(root->left, &ans, &count0, &val0);
        // 再遍历右子树
        recuriveCal(root->right, &ans, &count1, &val1);

        // 后遍历当前节点
        // 总节点数量
        int count = 1;
        count += (count0 + count1);

        // 总节点之和
        int val = root->val;
        val += (val0 + val1);

        // 满足则累加
        if(root->val == (int)(val / count)){
            ans++;
        }

        return ans;    
    }

    /*
        递归遍历计算函数
        类似树的后续遍历
    */
    void recuriveCal(TreeNode *root, int *ans, int *count, int *val){
        // 递归结束条件
        if(root == NULL){
            return;
        }

        // 记录左右子树的情况
        int count0 = 0, val0 = 0;
        int count1 = 0, val1 = 0;

        

        // 树的后续遍历
        // 先遍历左子树
        recuriveCal(root->left, ans, &count0, &val0);
        // 再遍历右子树
        recuriveCal(root->right, ans, &count1, &val1);

        // 后遍历当前节点
        (*count) = ((count0 + count1) + 1);
        (*val) = ((val0 + val1) + root->val);

        if(root->val == (int)((*val) / (*count))){
            (*ans)++;
        }
    }
};