统计最高分的节点数目-python

leetCode第2049题 统计最高分的节点数目
链接：https://leetcode-cn.com/problems/count-nodes-with-the-highest-score

给你一棵根节点为 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 <= 105
parents[0] == -1
对于 i != 0 ，有 0 <= parents[i] <= n - 1
parents 表示一棵二叉树。

先搞清楚题目的意思，题目中用的是parents列表的值(parents[i])表示的i节点的父结点，这样就很巧妙的用列表表示了树的结构。
我们对照着来看一下，下标i表示的是节点，值表示的是父节点

  示例1中：parents = [-1,2,0,2,0]
  [-1,2,0,2,0] 值
  [ 0,1,2,3,4] 下标
  那么0节点值为-1，这个是根节点，那么我们要先找到0的子节点，
  才能完整的构建树，找值为0的元素，通过下标我们知道了2，4是子节点

  那么我们接下来找2，4节点的子节点，就该找值为2和4的元素
  parents中只有2的元素，其下标分别为1，3。这样就找到了2的孩子节点
  而且树也构建完了

这就是示例1中树的由来.

接下来我们需要搞懂题目中分数的计算方式.

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

一个节点的分数是把这个节点拿掉,剩余子数的数目的乘积.

把0节点拿掉我们来看看

拿掉0节点剩余两棵子树,左边3个节点,右边1个节点,那么分数就是3*1=3

拿掉2节点看看

剩了3棵子树,数目分别为1,1,2,分数为112=2

搞懂了分数这个概念后,关键是怎么用代码清晰的表示分数的求解.
当拿掉一个节点后,我们最多产生3个子树,看上面这个例子,最多不会超过3个子树,如果是根节点的话,不会超过两个.
这很好理解,拿掉一个节点之后,可能会产生左子树,右子树,加上这个节点本身是某一个树的左(右)子树,那么还会剩余一棵.
那么分数就是这三个值相乘,a$\times$b$\times$c (根节点另外讨论)
-------------------------------------------
关键是如何求a,b,c的值,
我们假设a表示左子树,b表示右子树,c为兄弟节点的那棵树
①先遍历左子树,求出节点数目
②然后遍历右子树,求出节点数目
③兄弟节点其实就没有必要遍历了,因为树的节点数目我们是知道的,就是列表的长度,我们拿掉了一个节点就是n-1,我们又求出了a和b,那么c就是n-1-a-b.
④score = a$\times$b$\times$c

## python3
class Solution:
    def countHighestScoreNodes(self, parents: [int]) -> int:
        n = len(parents)
        child = [[]for i in range(n)]
        for node,p in enumerate(parents): # 既取出下标又取出元素
            if p != -1:
                child[p].append(node)  ## 求出所有节点的孩子节点
        maxsore,cnt = 0,0
        def dfs(node:int)->int:
            score = 1
            size = n-1  ## 把这个节点去掉
            for eadh in child[node]:
                sz = dfs(eadh)  ## 依次搜索子树的节点数目，求出a,b
                score *= sz
                size -= sz  ## c的值
            if node != 0:  ## 非根节点才有c值
                score *= size
            nonlocal maxsore,cnt
            if score == maxsore:
                cnt += 1
            elif score > maxsore:
                maxsore,cnt = score,1
            return n-size  ## n-c就是a+b+1的值，也就是子树的节点数目（左子树+右子树+自身）
        dfs(0)
        return cnt