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

Use Map for TrieNode children (#27)

Browse Source
This commit is contained in:
Cyril Auburtin 2018-05-29 05:13:12 +02:00 committed by Oleksii Trekhleb
parent 30ae3230d1
commit b1a613e03e
2 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,35 +2,31 @@ export default class TrieNode {
constructor(character, isCompleteWord = false) {
this.character = character;
this.isCompleteWord = isCompleteWord;
this.children = {};
this.children = new Map();
}
getChild(character) {
if (!Object.prototype.hasOwnProperty.call(this.children, character)) {
return null;
}
return this.children[character];
return this.children.get(character);
}
addChild(character, isCompleteWord = false) {
if (!this.children[character]) {
this.children[character] = new TrieNode(character, isCompleteWord);
if (!this.children.has(character)) {
this.children.set(character, new TrieNode(character, isCompleteWord));
}
return this.children[character];
return this.children.get(character);
}
hasChild(character) {
return !!this.children[character];
return this.children.has(character);
}
suggestChildren() {
return Object.keys(this.children);
return [...this.children.keys()];
}
toString() {
let childrenAsString = Object.keys(this.children).toString();
let childrenAsString = this.suggestChildren().toString();
childrenAsString = childrenAsString ? `:${childrenAsString}` : '';
const isCompleteString = this.isCompleteWord ? '*' : '';

2
src/data-structures/trie/__test__/TrieNode.test.js
View File

@ -26,7 +26,7 @@ describe('TrieNode', () => {
expect(trieNode.getChild('a').toString()).toBe('a');
expect(trieNode.getChild('o').toString()).toBe('o');
expect(trieNode.getChild('b')).toBeNull();
expect(trieNode.getChild('b')).toBeUndefined();
});
it('should check if node has specific child', () => {