aboutsummaryrefslogtreecommitdiff
path: root/include/tt_treap.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/tt_treap.h')
-rw-r--r--include/tt_treap.h40
1 files changed, 40 insertions, 0 deletions
diff --git a/include/tt_treap.h b/include/tt_treap.h
new file mode 100644
index 0000000..793d536
--- /dev/null
+++ b/include/tt_treap.h
@@ -0,0 +1,40 @@
+#include <stdlib.h>
+#include <stdint.h>
+#include <xorshift64.h>
+
+#ifndef TT_TREAP_H
+#define TT_TREAP_H
+
+// ===================================================================
+// Types
+// ===================================================================
+
+enum TT_FLAG { TT_EXACT, TT_LOWERBOUND, TT_UPPERBOUND };
+
+typedef struct treap_node_s {
+ uint64_t key;
+ uint32_t weight;
+ struct treap_node_s *left, *right, *parent;
+ enum TT_FLAG flag;
+ uint8_t depth;
+ float value;
+} tt_entry_t;
+
+// ===================================================================
+// Globals
+// ===================================================================
+
+extern uint32_t tt_num_cached;
+
+// ===================================================================
+// Methods
+// ===================================================================
+
+int tt_init(void);
+void tt_free(void);
+
+tt_entry_t *tt_seek(uint64_t key);
+int tt_insert(const uint64_t key, const enum TT_FLAG flag,
+ const uint8_t depth, const float value);
+
+#endif