From 37658c7bd095c71714723276a1d28c6565108d90 Mon Sep 17 00:00:00 2001 From: Simon Gardling Date: Tue, 28 Jan 2025 22:40:16 -0500 Subject: [PATCH] add diagonal support --- src/board.rs | 8 +++++++- src/game.rs | 4 +--- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/src/board.rs b/src/board.rs index 4935e15..636c02e 100644 --- a/src/board.rs +++ b/src/board.rs @@ -105,7 +105,13 @@ impl Board { .map(|range| range.into_iter().map(|j| (i, j)).collect()), ); - // TODO! add diagonals to `chains` here + chains.extend( + split_from(0, BOARD_SIZE - 1, i) + .into_iter() + .zip(split_from(0, BOARD_SIZE - 1, j)) + .map(|(i_vec, j_vec)| i_vec.into_iter().zip(j_vec)) + .map(|x| x.into_iter().map(|(i, j)| (i, j)).collect()), + ); let mut captured: usize = 0; diff --git a/src/game.rs b/src/game.rs index 7a4afc1..0a11e26 100644 --- a/src/game.rs +++ b/src/game.rs @@ -51,11 +51,9 @@ impl Game { println!("{}", self); - // return; - self.step(i); - // std::thread::sleep(Duration::from_millis(200)); + std::thread::sleep(Duration::from_millis(200)); } } }