Noratrieb/monte-carlo-tree-search
1
0
Fork 0
mirror of https://github.com/Noratrieb/monte-carlo-tree-search.git synced 2026-01-14 07:15:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

:(

Browse source
This commit is contained in:
nora 2022-01-30 18:38:54 +01:00
parent feac668bb8
commit 79c86e663f
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/lib.rs
View file

@ -17,8 +17,9 @@ pub trait PlayerState: Eq + Copy {
pub trait GameState: Clone + std::fmt::Debug {
type Player: PlayerState + std::fmt::Debug;
type NextStates: ExactSizeIterator<Item = Self>;
fn next_states(&self) -> Box<dyn ExactSizeIterator<Item = Self>>;
fn next_states(&self) -> Self::NextStates;
fn status(&self) -> Status<Self::Player>;
@ -46,7 +47,7 @@ mod mcts {
impl<'tree, S: GameState> Node<'tree, S> {
fn new(state: S, player: S::Player, alloc: &'tree Bump) -> Node<'tree, S> {
Node {
Self {
state,
player,
visited: Cell::new(0),
@ -273,7 +274,9 @@ pub mod tic_tac_toe {
impl GameState for Board {
type Player = Player;
fn next_states(&self) -> Box<dyn ExactSizeIterator<Item = Self>> {
type NextStates = std::vec::IntoIter<Self>;
fn next_states(&self) -> Self::NextStates {
let state_iter = self
.board
.iter()
@ -290,7 +293,7 @@ pub mod tic_tac_toe {
.collect::<Vec<_>>()
.into_iter();
Box::new(state_iter)
state_iter
}
fn status(&self) -> Status<Player> {