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 == parents.length
2 <= n <= 105
parents[0] == -1
对于 i != 0 ，有 0 <= parents[i] <= n - 1
parents 表示一棵二叉树。

代码

class Solution {
public:
    long maxScore = 0;
	int ans = 0;
	int n;


	int countHighestScoreNodes(vector<int>& parents) {
		n = parents.size();
		vector<int> left(n,-1);
		vector<int> right(n, -1);

		for (int i = 1; i < n; ++i)
		{
			if (left[parents[i]] == -1)
			{
				left[parents[i]] = i;
			}
			else
			{
				right[parents[i]] = i;
			}
		}
		
		dfs(0, left, right);

		return ans;
	}
	
	int dfs(int node, vector<int>& left, vector<int>& right)
	{
		if (node == -1)
		{
			return 0;
		}
		int left_node = dfs(left[node], left, right);
		int right_node = dfs(right[node], left, right);
		int remain = n - left_node - right_node - 1;

		long score = help(left_node) * help(right_node) * help(remain);
		if (score == maxScore)
		{
			ans++;
		}
		else if (score > maxScore)
		{
			maxScore = score;
			ans = 1;
		}

		return left_node + right_node + 1;

	}

	long help(int count)
	{
		return count == 0 ? 1 : count;
	}
};

基本思路

使用dfs求出当前节点的左子树和右子树的节点数，再用总节点数减去其左右子树的节点数及其本身，可得父节点链接的子树的节点数，三个子树相乘可得当前节点的分数。