2049. 统计最高分的节点数目

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-nodes-with-the-highest-score
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

import java.util.*;

class Solution {

    public int countHighestScoreNodes(int[] parents) {
        int n = parents.length;

        int[] in = new int[n];
        for (int i = 0; i < n; ++i) {
            if (parents[i] != -1) {
                in[parents[i]]++;
            }
        }

        Node[] nodes = new Node[n];
        Queue<Integer> queue = new LinkedList<>();
        for (int i = 0; i < n; ++i) {
            if (in[i] == 0) {
                queue.offer(i);
            }
            nodes[i] = new Node();
        }

        while (!queue.isEmpty()) {
            int child = queue.poll();
            int parent = parents[child];
            if (parent != -1) {
                nodes[parent].nodeNum += nodes[child].nodeNum;
                nodes[parent].children.add(child);
                in[parent]--;
                if (in[parent] == 0) {
                    queue.offer(parent);
                }
            }
        }

        long maxScore = 0;
        int cnt = 0;

        for (int i = 0; i < n; ++i) {
            int parent = parents[i];
            long multiply = parent == -1 ? 1 : (n - nodes[i].nodeNum);
            for (int child : nodes[i].children) {
                multiply *= nodes[child].nodeNum;
            }
            if (maxScore == multiply) {
                cnt++;
            } else if (multiply > maxScore) {
                maxScore = multiply;
                cnt = 1;
            }
        }

        return cnt;
    }
}

class Node {
    int nodeNum = 1;
    List<Integer> children = new ArrayList<>();
}