This commit is contained in:
Simon Gardling 2025-01-30 16:33:10 -05:00
parent 472141533b
commit 73d24e7748
Signed by: titaniumtown
GPG Key ID: 9AB28AC10ECE533D

View File

@ -1,5 +1,3 @@
use std::collections::HashMap;
use crate::{
agent::Agent,
board::{Board, BOARD_SIZE},
@ -49,20 +47,20 @@ impl Move {
}
}
fn problem_space(board: &Board, piece: Piece) -> Vec<Move> {
fn problem_space(board: &Board, color: Piece) -> Vec<Move> {
(0..BOARD_SIZE)
.flat_map(|i| {
(0..BOARD_SIZE)
.map(|j| (i, j))
.collect::<Vec<(usize, usize)>>()
})
.flat_map(|(i, j)| board.what_if(i, j, piece).map(|x| (i, j, x)))
.map(|(i, j, (b, c))| Move {
.flat_map(|(i, j)| board.what_if(i, j, color).map(|x| (i, j, x)))
.map(|(i, j, (board, captured))| Move {
i,
j,
captured: c,
color: piece,
board: b,
captured,
color,
board,
next_move: Vec::new(),
})
.collect()