tree-multimap-typed 2.4.1 → 2.4.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tree-multimap-typed",
-  "version": "2.4.1",
+  "version": "2.4.2",
   "description": "TreeMultiMap - A sorted map that allows multiple values per key",
   "browser": "dist/umd/tree-multimap-typed.min.js",
   "umd:main": "dist/umd/tree-multimap-typed.min.js",
@@ -217,6 +217,6 @@
     "typescript": "^4.9.5"
   },
   "dependencies": {
-    "data-structure-typed": "^2.4.1"
+    "data-structure-typed": "^2.4.2"
   }
 }

package/src/data-structures/trie/trie.ts

@@ -564,19 +564,17 @@ export class Trie<R = any> extends IterableElementBase<string, R> {
     const words: string[] = [];
     let found = 0;
 
-    function dfs(node: TrieNode, word: string) {
-      for (const char of node.children.keys()) {
-        const charNode = node.children.get(char);
-        if (charNode !== undefined) {
-          dfs(charNode, word.concat(char));
-        }
+    const dfs = (node: TrieNode, word: string): void => {
+      for (const [char, childNode] of node.children) {
+        if (found >= max) return;
+        dfs(childNode, word + char);
       }
       if (node.isEnd) {
-        if (found > max - 1) return;
+        if (found >= max) return;
         words.push(word);
         found++;
       }
-    }
+    };
 
     let startNode = this.root;