Added trie

package.json

@@ -43,6 +43,7 @@
     "eslint-plugin-standard": "^3.0.1",
     "husky": "^0.14.3",
     "mocha": "^5.1.1",
-    "nyc": "^11.7.1"
+    "nyc": "^11.7.1",
+    "random-words": "^1.1.0"
   }
 }

src/index.js

@@ -6,12 +6,14 @@ import CircularList from './circular-linked-list.js';
 import Stack from './stack.js';
 import Queue from './queue.js';
 import BST from './binary-search-tree.js';
+import Trie from './trie.js';
 
 export {
   List,
   DoublyList,
   CircularList,
   Stack,
   Queue,
-  BST
+  BST,
+  Trie
 };

src/trie.js

@@ -0,0 +1,81 @@
+'use strict';
+
+class Node {
+  constructor (value = null, isCompleteWord = false) {
+    this.value = value;
+    this.children = new Map();
+    this.isCompleteWord = isCompleteWord;
+  }
+
+  /**
+   * @param {char} value
+   * @param {Node} node
+   */
+  setChild (value, node) {
+    this.children.set(value, node);
+  }
+
+  /**
+   * @param  {char} value
+   * @return {Node}
+   */
+  getChild (value) {
+    return this.children.get(value);
+  }
+}
+
+export default class Trie {
+  constructor () {
+    this.root = new Node();
+  }
+
+  /**
+  * @param {string} word
+  */
+  addWord (word) {
+    if (!this.isValidWord(word)) {
+      throw Error('Invalid Word');
+    }
+
+    let current = this.root;
+
+    Array.from(word).forEach((letter) => {
+      if (!current.getChild(letter)) {
+        current.setChild(letter, new Node(letter));
+      }
+
+      current = current.getChild(letter);
+    });
+
+    current.isCompleteWord = true;
+  }
+
+  /**
+  * @return {Array<string>}
+  */
+  getWords () {
+    let words = [];
+
+    const getWords = (node, wordFragment) => {
+      node.children.forEach((child) => {
+        if (child.isCompleteWord) {
+          words.push(wordFragment + child.value);
+        }
+
+        getWords(child, wordFragment + child.value);
+      });
+    };
+
+    getWords(this.root, '');
+
+    return words;
+  }
+
+  /**
+  * @param  {string}  word
+  * @return {Boolean}
+  */
+  isValidWord (word) {
+    return (typeof (word) === 'string' || word instanceof String) && word.match(/^[A-z]+$/);
+  }
+}

test/trie-test.js

@@ -0,0 +1,29 @@
+'use strict';
+
+import {Trie} from '../src/index.js';
+
+const randomWords = require('random-words');
+const expect = require('chai').expect;
+
+describe('testing trie', function () {
+  it('test adding values', function () {
+    let trie = new Trie();
+    let words = randomWords(50);
+
+    words.forEach((word) => {
+      trie.addWord(word);
+    });
+
+    expect(trie.getWords()).to.include.members(words);
+  });
+
+  it('test adding invalid values', function () {
+    let trie = new Trie();
+
+    let invalidInputs = ['', 123, '$%^&', '123'];
+
+    invalidInputs.forEach((input) => {
+      expect(trie.addWord.bind(trie, input)).to.throw(Error, 'Invalid Word');
+    });
+  });
+});