给你一棵根节点为 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