JAVA练习207-统计最高分的节点数目

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

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

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

示例 1:

输入：parents = [-1,2,0,2,0]
输出：3
解释：
- 节点 0 的分数为：3 * 1 = 3
- 节点 1 的分数为：4 = 4
- 节点 2 的分数为：1 * 1 * 2 = 2
- 节点 3 的分数为：4 = 4
- 节点 4 的分数为：4 = 4
最高得分为 4 ，有三个节点得分为 4 （分别是节点 1，3 和 4 ）。

示例 2：

输入：parents = [-1,2,0]
输出：2
解释：
- 节点 0 的分数为：2 = 2
- 节点 1 的分数为：2 = 2
- 节点 2 的分数为：1 * 1 = 1
最高分数为 2 ，有两个节点分数为 2 （分别为节点 0 和 1 ）。

提示：

• n == parents.length
• 2 <= n <= 10^5
• parents[0] == -1
• 对于 i != 0 ，有 0 <= parents[i] <= n - 1
• parents 表示一棵二叉树。

分析：

方法：DFS

删除掉某一节点后，最多会产生三棵非空子树，两棵都为该节点的左右子树，一棵为原数去掉以该节点为根的树的子树，因此只需要求这三棵数的节点数相乘即可。我们可以采用深度优先（DFS）的方式进行遍历，先将数组还原成一棵树，我们再依次遍历它的节点，左右子树的节点数可以深度递归获得，另一棵树的节点可以通过总节点数减去左右节点的节点数再减1获得。

时间复杂度：O(n)        n 为节点数
空间复杂度：O(n)  

class Solution {

    //节点集合数组，索引代表节点，集合代表它的子节点
    List<Integer>[] nodes;
    //总节点数，最高得分节点数
    int len = 0, ans = 0;
    //最高得分
    long max = 0;

    public int countHighestScoreNodes(int[] parents) {
        //节点数为数组长度
        len = parents.length;
        //节点集合数组初始化
        nodes = new List[len];
        for(int i = 0; i < len; ++i){
            nodes[i] = new ArrayList<>();
        }
        //赋值
        for(int i = 1; i < len; ++i){
            nodes[parents[i]].add(i);
        }
        //深度遍历
        dfs(0);
        //返回结果
        return ans;
    }

    public int dfs(int n){
        //分数
        long score = 1;
        //原树去掉以该节点为根的树的子树的节点数
        int ori = n == 0?len:len-1;
        //遍历子节点
        for(int son: nodes[n]){
            int s = dfs(son);
            score *= s;
            ori -= s;
        }
        //乘上剩余节点数
        score *= ori;
        //判断是否为最大分数
        //大于最大分数，更新最大分数，重置答案
        if(score > max){
            max = score;
            ans = 1;
        }
        //等于最大分数，答案数+1
        else if(score == max){
            ans++;
        }
        //返回左右子树加上当前节点的节点数
        return len-ori;
    }
}

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-nodes-with-the-highest-score