lib: trie implementation

2026-10-27 20:58:22 +00:00 · 2026-07-12 13:28:13 -03:00
parent e91d82cf09
commit e630b7fa11
2 changed files with 340 additions and 0 deletions
@@ -0,0 +1,35 @@
+#ifndef _TRIE_H
+#define _TRIE_H
+
+#include <stdint.h>
+
+#define TRIE_BITS 6
+#define TRIE_NODE_COUNT (1 << TRIE_BITS)
+
+typedef struct trie_node trie_node_t;
+struct trie_node {
+	uint8_t count;
+	union {
+		trie_node_t *nodes[TRIE_NODE_COUNT];
+		void *items[TRIE_NODE_COUNT];
+	};
+};
+
+typedef struct {
+	uint8_t height;
+	union {
+		void *root_item;
+		trie_node_t *root;
+	};
+} trie_t;
+
+typedef void (trie_iterate_fn_t)(void *item);
+
+void trie_init(trie_t *trie);
+int trie_insert(trie_t *trie, uint64_t key, void *value);
+int trie_lookup(trie_t *trie, uint64_t key, void **ret);
+int trie_remove(trie_t *trie, uint64_t key);
+void trie_truncate(trie_t *trie, uint64_t max_key);
+void trie_iterate(trie_t *trie, uint64_t min_key, uint64_t max_key, trie_iterate_fn_t);
+
+#endif