统计最高分的节点数目-CSDN博客

本文链接：https://blog.csdn.net/weixin_44218238/article/details/123455142

题目
给你一棵根节点为 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;
        
    }
};