【LeetCode 6057】 统计值等于子树平均值的节点数（周赛）

题目描述

给你一棵二叉树的根节点 root ，找出并返回满足要求的节点数，要求节点的值等于其 子树 中值的 平均值 。
注意：

• n 个元素的平均值可以由 n 个元素 求和 然后再除以 n ，并 向下舍入 到最近的整数。
• root 的 子树 由 root 和它的所有后代组成。
  示例1
  

输入：root = [4,8,5,0,1,null,6]
输出：5
解释：
对值为 4 的节点：子树的平均值 (4 + 8 + 5 + 0 + 1 + 6) / 6 = 24 / 6 = 4 。
对值为 5 的节点：子树的平均值 (5 + 6) / 2 = 11 / 2 = 5 。
对值为 0 的节点：子树的平均值 0 / 1 = 0 。
对值为 1 的节点：子树的平均值 1 / 1 = 1 。
对值为 6 的节点：子树的平均值 6 / 1 = 6 。

示例 2：

解题思路

这道题考察的是对二叉树的遍历操作(递归和非递归)，比较麻烦的是计算当前节点所有子树（包括自己）的均值！
然后再返回二叉树中节点的值 和 其子树均值相等的节点数。
1.用递归先序遍历 ，计算子树平均值
2.用非递归先序遍历(栈实现) ，判断节点的值和 其子树均值是否相等，相等就+1。

int account=0,nodesum=0;//全局变量
int sum(TreeNode* root){//递归先序遍历就算节点子树的平均值
	if(!root) return 0;
	account+=root->val;
	nodesum++;
	sum(root->left);
	sum(root->right);

	return account/nodesum;
}
int averageOfSubtree(TreeNode* root) {//非递归先序遍历 判断是否相等 并返回答案
	stack<TreeNode*> tmp;
	int result=0;
	if(!root) return 0;
	tmp.push(root);
	while(!tmp.empty()){
		TreeNode *P=tmp.top();
		tmp.pop();
		account=0,nodesum=0;//这里需要赋值为0
		if(P->val==sum(P))
			result++;
		if(P->right)
			tmp.push(P->right);
		if(P->left)
			tmp.push(P->left);
	}
	return result;
}