來源：力扣（LeetCode）

連結：https://leetcode-cn.com/problems/count-nodes-with-the-highest-score

題目描述

給你一棵根節點為 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⁵

parents[0] == -1

對於 i != 0 ，有 0 <= parents[i] <= n - 1

parents 表示一棵二叉樹。

解題思路

首先parant陣列儲存形式對於數來說不直觀，很難進行操作，所以遍歷parant陣列，利用鄰接連結串列建立一顆二叉樹。然後使用二叉樹的後續遍歷，分別將左右子樹和分數全求出來，進行比較和計數，需要注意int相乘可能會越界，需要使用long。

程式碼展示

class Solution {
public:
    vector<vector<int>> mvviTree;
    long long miMax;
    int miCount;
    void dfs(int root, int &count)
    {
        int iCount = 0;
        long long iCount1 = 1;
        count = 1;
        for(auto iter:mvviTree[root])
        {
            dfs(iter, iCount);
            count += iCount;
            iCount1 *= iCount;
        }
        if(root)
            iCount1 *= (mvviTree.size() - count);
        if(iCount1 > miMax)
        {
            miMax = iCount1;
            miCount = 1;
        }
        else if(iCount1 == miMax)
        {
            miCount++;
        }
    }
    int countHighestScoreNodes(vector<int>& parents) {
        int n = parents.size();
        miCount = 0;
        miMax = 0;
        mvviTree.resize(n, vector<int>());
        int iCount = 0;
        for(int i = 1; i < n; i++)
        {
            mvviTree[parents[i]].push_back(i);
        }
        dfs(0, iCount);
        return miCount;

    }
};

執行結果