titaniumtown/othello
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve split_from

Browse Source
This commit is contained in:
Simon Gardling 2025-01-27 22:19:35 -05:00
parent 58b84fa3ac
commit be8f97d4f1
Signed by: titaniumtown
GPG Key ID: 9AB28AC10ECE533D
2 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/misc.rs
View File

@ -1,16 +1,10 @@
pub fn split(min: usize, max: usize, x: usize) -> Vec<Vec<usize>> {
pub fn split_from(min: usize, max: usize, x: usize) -> Vec<Vec<usize>> {
if min > x || x > max {
return Vec::new();
}
let index = x - min;
let collected = (min..=max).collect::<Vec<usize>>();
let split = collected.split_at(index);
let mut one = split.0.to_vec();
one.reverse();
let mut two = split.1.to_vec();
two.remove(0);
vec![one, two]
let x_lower = x.checked_sub(1).unwrap_or(min);
let x_upper = (x + 1).min(max);
vec![(min..=x_lower).rev().collect(), (x_upper..=max).collect()]
}
#[cfg(test)]
@ -19,6 +13,6 @@ mod test {
#[test]
fn split_test() {
assert_eq!(split(0, 6, 2), vec![vec![1, 0], vec![3, 4, 5, 6]]);
assert_eq!(split_from(0, 6, 2), vec![vec![1, 0], vec![3, 4, 5, 6]]);
}
}

6
src/repr.rs
View File

@ -1,6 +1,6 @@
use std::fmt;
use crate::misc::split;
use crate::misc::split_from;
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Piece {
@ -80,12 +80,12 @@ impl Board {
return;
};
let mut chains: Vec<Vec<(usize, usize)>> = split(0, BOARD_SIZE - 1, i)
let mut chains: Vec<Vec<(usize, usize)>> = split_from(0, BOARD_SIZE - 1, i)
.into_iter()
.map(|range| range.into_iter().map(|i| (i, j)).collect())
.collect();
chains.append(
&mut split(0, BOARD_SIZE - 1, j)
&mut split_from(0, BOARD_SIZE - 1, j)
.into_iter()
.map(|range| range.into_iter().map(|j| (i, j)).collect())
.collect::<Vec<Vec<(usize, usize)>>>(),