Update 2049-count-nodes-with-the-highest-score.js

Author: everthis

2049-count-nodes-with-the-highest-score.js

@@ -24,3 +24,35 @@ const countHighestScoreNodes = function(parents) {
   const maxKey = Math.max(...Object.keys(hash))
   return hash[maxKey]
 };
+
+// another
+
+/**
+ * @param {number[]} parents
+ * @return {number}
+ */
+const countHighestScoreNodes = function(parents) {
+  const n = parents.length, hash = {}, graph = {}
+  for(let i = 1; i < n; i++) {
+    if(graph[parents[i]] == null) graph[parents[i]] = []
+    graph[parents[i]].push(i)
+  }
+
+  dfs(0)
+  const mk = Math.max(...Object.keys(hash))
+  return hash[mk]
+  
+  function dfs(i) {
+    let num = 0, prod = 1
+    for(const e of (graph[i] || []) ) {
+      const tmp = dfs(e)
+      num += tmp
+      prod *= tmp
+    }
+    
+    if(n - 1 - num > 0) prod *= (n - 1 - num)
+    hash[prod] = (hash[prod] || 0) + 1
+
+    return num + 1
+  }
+};