update: total words in a Trie

Author: ashokdey

src/_DataStructures_/Trees/Trie/total-words-in-trie/index.js

@@ -0,0 +1,23 @@
+// eslint-disable-next-line no-unused-vars
+const Trie = require('../index');
+
+function totalWords(root) {
+  let result = 0;
+  if (root.isEndOfWord) {
+    result += 1;
+  }
+  for (let i = 0; i < 26; i += 1) {
+    if (root.children[i] !== null) {
+      result += totalWords(root.children[i]);
+    }
+  }
+  return result;
+}
+
+// const words = ['bed', 'ball', 'apple', 'java', 'javascript'];
+// const trie = new Trie();
+
+// words.forEach(word => trie.insert(word));
+// console.log(totalWords(trie.root));
+
+module.exports = totalWords;