1 files changed, 52 insertions, 0 deletions
diff --git a/modules/core/mcts.h b/modules/core/mcts.h
new file mode 100644
index 0000000..8a8f5ea
--- /dev/null
+++ b/modules/core/mcts.h
@@ -0,0 +1,52 @@
+#pragma once
+
+#include "tree.h"
+
+namespace saw {
+template<typename T>
+class mcts_tree {
+private:
+	struct value {
+		uint64_t numerater;
+		uint64_t denominater;
+		T state;
+
+		value() = default;
+		value(T st):
+			numerater{0},
+			denominater{0},
+			state{std::move(st)}
+		{}
+	};
+
+	tree_container<value, mcts_tree<T>> data_;
+public:
+	mcts_tree() = default{
+		data_.add(value{});
+	}
+
+	mcts_tree(T state){
+		data_.add(value{std::move(state)});
+	}
+
+	size_t size() const {
+		return data_.size() - 1;
+	}
+
+	T& get_state(){
+		return data_.at(0).get_value().state;
+	}
+
+	const T& get_state() const {
+		return data_.at(0).get_value().state;
+	}
+
+	mcts_tree<T>& get_tree(size_t i){
+		return data_.at(i+1).get_tree();
+	}
+
+	const mcts_tree<T>& get_tree(size_t i) const {
+		return data_.at(i+1).get_tree();
+	}
+};
+}