Noratrieb/monte-carlo-tree-search
1
0
Fork 0
mirror of https://github.com/Noratrieb/monte-carlo-tree-search.git synced 2026-01-14 15:25:09 +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 { pub trait GameState: Clone + std::fmt::Debug {
type Player: PlayerState + 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>; fn status(&self) -> Status<Self::Player>;
@ -46,7 +47,7 @@ mod mcts {
impl<'tree, S: GameState> Node<'tree, S> { impl<'tree, S: GameState> Node<'tree, S> {
fn new(state: S, player: S::Player, alloc: &'tree Bump) -> Node<'tree, S> { fn new(state: S, player: S::Player, alloc: &'tree Bump) -> Node<'tree, S> {
Node { Self {
state, state,
player, player,
visited: Cell::new(0), visited: Cell::new(0),
@ -273,7 +274,9 @@ pub mod tic_tac_toe {
impl GameState for Board { impl GameState for Board {
type Player = Player; 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 let state_iter = self
.board .board
.iter() .iter()
@ -290,7 +293,7 @@ pub mod tic_tac_toe {
.collect::<Vec<_>>() .collect::<Vec<_>>()
.into_iter(); .into_iter();
Box::new(state_iter) state_iter
} }
fn status(&self) -> Status<Player> { fn status(&self) -> Status<Player> {