Fix method Trie::doesWordExist() (trekhleb#175)

Oleg · trekhleb · commit 392cd9806dcc · 2018-08-21T23:00:11.000+03:00
Method Trie::doesWordExist() return `true` when word is complete otherwise `false`
diff --git a/src/data-structures/trie/Trie.js b/src/data-structures/trie/Trie.js
@@ -43,7 +43,9 @@ export default class Trie {
    * @return {boolean}
    */
   doesWordExist(word) {
-    return !!this.getLastCharacterNode(word);
+    const lastCharacter = this.getLastCharacterNode(word);
+
+    return !!lastCharacter && lastCharacter.isCompleteWord;
   }
 
   /**
diff --git a/src/data-structures/trie/__test__/Trie.test.js b/src/data-structures/trie/__test__/Trie.test.js
@@ -45,7 +45,8 @@ describe('Trie', () => {
     trie.addWord('caption');
 
     expect(trie.doesWordExist('cat')).toBe(true);
-    expect(trie.doesWordExist('cap')).toBe(true);
+    expect(trie.doesWordExist('cats')).toBe(true);
+    expect(trie.doesWordExist('cap')).toBe(false);
     expect(trie.doesWordExist('call')).toBe(false);
   });
 });
