文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給你一棵根節點為 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 表示一棵二叉樹。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/count-nodes-with-the-highest-score
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 建圖，dfs，自底向上求子樹節點數量
• dfs，求取每個節點的得分

class Solution {vector<vector<int>> g;vector<long long> score;long long ans = 0, n; public:int countHighestScoreNodes(vector<int>& parents) {n = parents.size();g.resize(n);score.resize(n);for(int i = 1; i < n; ++i)g[parents[i]].push_back(i);vector<int> node(n, 0);dfs(0, node);dfs1(0, node);int ct = 0;for(auto s : score){if(s == ans)ct++;}return ct;}int dfs(int x, vector<int>& node){node[x]++;for(auto y : g[x])node[x] += dfs(y, node);return node[x];}void dfs1(int x, vector<int>& node){for(auto y : g[x])dfs1(y, node);if(g[x].size()==0) score[x] = n-1;else if(g[x].size()==1){int n1 = node[g[x][0]];score[x] = 1LL*n1*((n-1-n1)==0? 1 :(n-1-n1));}else{int n1 = node[g[x][0]];int n2 = node[g[x][1]];score[x] = 1LL*n1*n2*((n-1-n1-n2)==0? 1 : (n-1-n1-n2));}ans = max(ans, score[x]);} };

276 ms 132.7 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 2049. 统计最高分的节点数目（DFS）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。