题意
(题目的说法虽然没什么错误,但我还是觉得十分难理解,所以直接看题目下面示例1的那张图会比较好)第一段就是说一个数组parents
代表每个节点的父节点是谁,0节点一定是根节点,所以数组第一个数一定为-1。然后这棵树一定会是二叉树。
第二段就是题目十分绕口的地方,这个二叉树上每个节点都有它自己相对应的分数,这个分数的计算方法:把当前节点删去,原本的二叉树有可能被分为一个或两个或三个二叉树,删除叶子节点后,二叉树还是二叉树,但若是删除有左右孩子的节点,二叉树就可以分出多个二叉树,题目要求的就是剩余(一个或多个)子树节点个数的乘积(看题目示例的图会很好理解的)
求有最高得分的节点个数。
思路
这里有一个能够建立二叉树的数组,首先就是要建树,因为节点不多且是二叉树,用二维的vector就行,然后0节点一定是根节点建树可以不遍历它。
一个节点被删除后,它的分数可以分为三个区域的乘积,它的左子树的结点个数、右子树的结点个数,以及父节点的“连通块”个数,这三个区域需要非空才能进行相乘,两个子树的个数用深搜,父节点区域就是总节点个数减去当前节点的子节点个数(包含其自身),深搜过程统计最大值和最大值出现的个数。
这里的乘积最大值不用用int,反正我wa了一发(惨)
代码
class Solution {
public:
int ans = 0;
long long maxFen = 0;
int n;
vector<vector<int>> trie;
int dfs(int x){
long long fen = 1;
int zishu = 1;
for(int y:trie[x]){
int z = dfs(y);
fen *= z;
zishu += z;
}
if(x) fen *= n-zishu;
if(fen==maxFen)++ans;
else if(fen>maxFen){
maxFen = fen;
ans = 1;
}
return zishu;
}
int countHighestScoreNodes(vector<int>& parents) {
this->n = parents.size();
this->trie = vector<vector<int>>(n);
for(int i = 1; i < n; ++i){
trie[parents[i]].emplace_back(i);
}
dfs(0);
return ans;
}
};