LeetCode刷题日记2022-3-11/2049. 统计最高分的节点数目-DFS

题目描述

给你一棵根节点为 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 表示一棵二叉树。

题解思路

由题目可知 我们计算一个节点的得分 是由其左子树left 和 右子树 right的大小得到的

而score得得分 是由

• leftright（n-left-right-1）计算

其中

• left,right为该节点的左右子树

• （n-left-right-1）为断开该节点左右子树及根节点得节点数

由此我们可以建立children 来记录节点得子节点 然后遍历节点 同时更新最大得分及其数量

题解代码

class Solution:
    def countHighestScoreNodes(self, parents: List[int]) -> int:
        n=len(parents)
        children=defaultdict(list)
        for node,p in enumerate(parents):
            if p!=-1:
                children[p].append(node)
        
        maxscore,cnt=0,0

        def dfs(node):
            left=dfs(children[node][0]) if children[node] else 0
            right=dfs(children[node][1]) if len(children[node])==2 else 0
            nonlocal maxscore,cnt
            score=max(1,(n-left-right-1))*max(1,left)*max(1,right)
            if score>maxscore:
                maxscore=score
                cnt=1
            elif score==maxscore:
                cnt+=1
            return left+right+1
        dfs(0)

        return cnt