hxx/javascript-algorithms
1
0
Fork 0
mirror of https://github.moeyy.xyz/https://github.com/trekhleb/javascript-algorithms.git synced 2024-12-26 07:01:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix TrieNode.addChild so substrings get marked as complete words (#177)

Browse Source
This commit is contained in:
Kevin Brewer 2018-08-25 05:27:44 -05:00 committed by Oleksii Trekhleb
parent 872a38fac6
commit eac3e81a21
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/data-structures/trie/TrieNode.js
View File

@ -29,7 +29,10 @@ export default class TrieNode {
this.children.set(character, new TrieNode(character, isCompleteWord)); this.children.set(character, new TrieNode(character, isCompleteWord));
} }
return this.children.get(character); const childNode = this.children.get(character);
childNode.isCompleteWord = childNode.isCompleteWord || isCompleteWord;
return childNode;
} }
/** /**

3
src/data-structures/trie/__test__/Trie.test.js
View File

@ -41,11 +41,14 @@ describe('Trie', () => {
trie.addWord('cat'); trie.addWord('cat');
trie.addWord('cats'); trie.addWord('cats');
trie.addWord('carpet');
trie.addWord('car'); trie.addWord('car');
trie.addWord('caption'); trie.addWord('caption');
expect(trie.doesWordExist('cat')).toBe(true); expect(trie.doesWordExist('cat')).toBe(true);
expect(trie.doesWordExist('cats')).toBe(true); expect(trie.doesWordExist('cats')).toBe(true);
expect(trie.doesWordExist('carpet')).toBe(true);
expect(trie.doesWordExist('car')).toBe(true);
expect(trie.doesWordExist('cap')).toBe(false); expect(trie.doesWordExist('cap')).toBe(false);
expect(trie.doesWordExist('call')).toBe(false); expect(trie.doesWordExist('call')).toBe(false);
}); });