exhaustive-tree-search

uriva · Jan 3, 2025 · 5fbbe20 · 5fbbe20
1 parent 3b5da9b
commit 5fbbe20
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/tree.ts b/src/tree.ts
@@ -20,3 +20,12 @@ export const findInTree =
     }
     return null;
   };
+
+export const findInTreeExhaustive =
+  <T>(predicate: (t: T) => boolean, children: (t: T) => T[]) => (t: T): T[] => {
+    const results = (predicate(t)) ? [t] : [];
+    for (const child of children(t)) {
+      results.push(...findInTreeExhaustive(predicate, children)(child));
+    }
+    return results;
+  };