LeetCode 2049. 统计最高分的节点数目--树的遍历

2049. 统计最高分的节点数目
      给你一棵根节点为 0 的 二叉树 ，它总共有 n 个节点，节点编号为 0 到 n - 1 。同时给你一个下标从 0 开始的整数数组 parents 表示这棵树，其中 parents[i] 是节点 i 的父节点。由于节点 0 是根，所以 parents[0] == -1 。

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

请你返回有 最高得分 节点的 数目 。

示例 1:

example-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：
  

example-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 表示一棵二叉树。

题解

就是统计下每个节点的左子树和右子树大小就行了。

AC代码

class Solution {
public:
    typedef long long ll;
    ll L[100005],R[100005];//记录每个节点的左子树和右子树大小
    int L_child[100005],R_child[100005];//记录每个节点的左孩子和右孩子节点
    int dfs(int root)//dfs遍历每个节点的左子树大小和右子树大小,不包括自己本身
    {
        if(root==-1)return 0;
        L[root] = dfs(L_child[root]);
        R[root] = dfs(R_child[root]);
        return L[root] + R[root] + 1;//+1表示加上根节点本身
    }
    int countHighestScoreNodes(vector<int>& parents) 
    {
        //没有声明节点和父节点的关系，所以这里认为先出现的就是左子树，后出现的就是右子树
        memset(L_child,-1,sizeof(L_child));
        memset(R_child,-1,sizeof(R_child));
        for(int i=0;i<parents.size();i++)
        {
            int u = parents[i];
            if(u!=-1)
            {
                if(L_child[u]==-1)
                L_child[u] = i;
                else
                R_child[u] = i;
            }
        }
        memset(L,0,sizeof(L));
        memset(R,0,sizeof(R));
        dfs(0);
        ll mx = 0;
        for(int i=0;i<parents.size();i++)
        {
            ll L_ans=1,R_ans=1,parent_ans=1;
            if(L[i]!=0)L_ans = L[i];
            if(R[i]!=0)R_ans = R[i];
            if(parents.size()-L[i]-R[i]-1!=0)parent_ans = parents.size()-L[i]-R[i]-1;
            mx = max(mx, L_ans*R_ans*parent_ans);
        }
        int res = 0;
        for(int i=0;i<parents.size();i++)
        {
            ll L_ans=1,R_ans=1,parent_ans=1;
            if(L[i]!=0)L_ans = L[i];
            if(R[i]!=0)R_ans = R[i];
            if(parents.size()-L[i]-R[i]-1!=0)parent_ans = parents.size()-L[i]-R[i]-1;
            if(L_ans*R_ans*parent_ans==mx)
            res++;
        }
        return  res;
    }
};