leetcode5980统计最高分的节点数目

给你一棵根节点为 0 的 二叉树 ，它总共有 n 个节点，节点编号为 0 到 n - 1 。同时给你一个下标从 0 开始的整数数组 parents 表示这棵树，其中 parents[i] 是节点 i 的父节点。由于节点 0 是根，所以 parents[0] == -1 。

一个子树的 大小 为这个子树内节点的数目。每个节点都有一个与之关联的 分数 。求出某个节点分数的方法是，将这个节点和与它相连的边全部 删除 ，剩余部分是若干个 非空 子树，这个节点的 分数 为所有这些子树 大小的乘积 。

请你返回有 最高得分 节点的 数目

dfs统计每个结点所在子树的结点个数cnt, 枚举每个结点，结果就是cnt[l] * cnt[r] * (n - cnt[l] - cnt[r] - 1)

#define N 200005
#define ll long long int
#define mod 1e9 + 7
#define lowbit(x) x&(-x)
class Solution {
public:
    int cnt[100005];
    vector<int> dp[100003];
    int dfs(int n){
	if(dp[n].size() == 0){
		//cnt[n] = 1;
		return 1;
		}
		for(auto x : dp[n]){
			cnt[n] += dfs(x);
		}
		return cnt[n];
	}
    int countHighestScoreNodes(vector<int>& a) {
        int n = a.size();
        for(int i = 0; i < n; i++)
            cnt[i] = 1;
        for(int i = 1; i < n; i++)
        	dp[a[i]].push_back(i);
        dfs(0);
        ll ans = 0;
        map<ll, int> mp;
        for(int i = 0; i < n; i++){
        	ll inf = 1, k = n;
        	for(int j = 0; j < dp[i].size(); j++){
        		inf *= cnt[dp[i][j]];
        		k -= cnt[dp[i][j]];
        	}
        	k--;
        	if(k) inf *= k;
        	mp[inf]++;
        	ans = max(ans, inf);
        }
        return mp[ans];
    }
};