leetcode 2265统计值等于子树平均值的节点数

2022来水第一篇博客

2022年5月12日，开始封校的第一天生活，在苏州街公寓下午蹭了所里两小时的高级算法课，学了一点调度算法。20：51刷了一道leetcode水题 ，接下来看看论文吧。

/**
 * 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 res = 0;
    vector<int>dfs(TreeNode* root) {
    
        vector<int> ans(2, 0);
        if(root == nullptr) return ans;
        vector<int> val_1 = dfs(root->left);
        vector<int> val_2 = dfs(root->right);

        ans[0] = val_1[0] + val_2[0] + root->val;
        ans[1] = val_1[1] + val_2[1] + 1;
        if (ans[0] / ans[1] == root->val) res += 1;
        return ans;
    }
    int averageOfSubtree(TreeNode* root) {
        vector<int> ans = dfs(root);
        // cout<< ans[0] <<" "<< ans[1] <<" " << res<<endl;
        return res;
        
    }
};