Find Mode in Binary Search Tree.js (codesONLY#70)

DEV-AB-03 · web-flow · commit 9df78e59c07d · 2021-10-20T12:44:45.000+05:30
diff --git a/DSA/DFS/Find Mode in Binary Search Tree.js b/DSA/DFS/Find Mode in Binary Search Tree.js
@@ -0,0 +1,28 @@
+/**
+ * Definition for a binary tree node.
+ * function TreeNode(val, left, right) {
+ *     this.val = (val===undefined ? 0 : val)
+ *     this.left = (left===undefined ? null : left)
+ *     this.right = (right===undefined ? null : right)
+ * }
+ */
+var findMode = function(r) {
+  const map = new Map()
+  const maxMap = new Map()
+  let max = -Infinity
+  
+  function dfs(root){
+    if(!root) return
+    map.set(root.val, (map.get(root.val) || 0) + 1)
+      if(map.get(root.val) >= max){
+        max = map.get(root.val)
+        if(!maxMap.get(max)) maxMap.set(max, new Set())
+        maxMap.get(max).add(root.val)
+      }    
+    dfs(root.left)
+    dfs(root.right)
+  } 
+  
+  dfs(r)
+  return [...maxMap.get(max)]
+};
