leetcode树之统计最高分的节点数目

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

给你一棵根节点为 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-1，减去删除的节点。

根节点例外，删除根节点只有两个子树

这样可以统计每个节点的左右子树，那么另外一个子树的数目就是n-1-左子树节点数-右子树节点数

class Solution:
    def countHighestScoreNodes(self, parents: List[int]) -> int:
        n = len(parents)
        # 缓存父节点和子节点的关系
        children = collections.defaultdict(list)
        for node, parent in enumerate(parents):
            if parent != -1:
                children[parent].append(node)
        self.maxScore = float("-inf")
        self.cnt = 0
        # 返回当前结点的节点数（当前节点 + 左右子节点数）
        def dfs(root):
            score = 1
            size = n - 1
            # 遍历当前结点，获取左右子树的节点数
            # 如果删除当前结点，那么分数应该是子树节点数之积
            for child in children[root]:
                sz = dfs(child)
                score *= sz
                size -= sz
            # 排除根节点，删除根节点只有左右两个子树
            if root != 0:
                score *= size
            # 统计分数及当前最大分数的个数
            # 如果分数等于最大分数，说明最大分数不止一个，这里加一即可
            # 否则更新最大分数，并重置数目为1
            if score == self.maxScore:
                self.cnt += 1
            elif score > self.maxScore:
                self.cnt = 1
                self.maxScore = score
            # size是减去左右子树之后剩余的节点数 - 1（size初始化为n-1，已经减去了当前节点）
            # 而当前节点的节点数，应该是左右子树节点数 + 1
            # 所以n-size就是当前节点的节点数
            return n - size
        dfs(0)
        return self.cnt