题目
给你一棵根节点为 0 的 二叉树 ,它总共有 n 个节点,节点编号为 0 到 n - 1 。同时给你一个下标从 0 开始的整数数组 parents 表示这棵树,其中 parents[i] 是节点 i 的父节点。由于节点 0 是根,所以 parents[0] == -1 。
一个子树的 大小 为这个子树内节点的数目。每个节点都有一个与之关联的 分数 。求出某个节点分数的方法是,将这个节点和与它相连的边全部 删除 ,剩余部分是若干个 非空 子树,这个节点的 分数 为所有这些子树 大小的乘积 。
请你返回有 最高得分 节点的 数目 。
链接:https://leetcode-cn.com/problems/count-nodes-with-the-highest-score
思路
后序遍历,思路和后序遍历求解二叉树的深度一致
需要将树的双亲表示法变为孩子表示法,便于进行搜索遍历
去掉该结点后,分为三部分
子问题变为:求解以某节点作为根节点的子树中节点的数目
总结点减去两棵子树的节点即为剩余部分的节点
代码
class Solution {
public:
unordered_map<int,vector<int>> mp;
int n;
long long max_score=0;
int ans=0;
int dfs(int root){ // 以root作为根节点的子树中节点的个数
long long score=1;
int sum=0;
for(int i=0;i<mp[root].size();i++){
int sub_size=dfs(mp[root][i]);
score*=sub_size;
sum+=sub_size;
}
if(root)
score*=(n-1-sum);
if(score==max_score)
ans++;
else if(score>max_score){
ans=1;
max_score=score;
}
return sum+1;
}
int countHighestScoreNodes(vector<int>& parents) {
n=parents.size();
for(int i=0;i<n;i++){
mp[parents[i]].push_back(i);
}
dfs(0);
return ans;
}
};