题目描述
给你一棵二叉树的根节点 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;
}