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

optimize FutureMoves::prune_unrelated

Browse Source
This commit is contained in:
Simon Gardling
2025-02-10 10:47:02 -05:00
parent 0bd8dec65d
commit dd783f0608
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/complexagent.rs
View File

@@ -223,15 +223,18 @@ impl FutureMoves {
stack.extend(self.arena[idx].children.iter().copied());
}
let mut new_arena = Vec::new();
let mut new_arena = Vec::with_capacity(retain.iter().filter(|x| **x).count());
let mut index_map = vec![None; self.arena.len()];
for (old_idx, _) in retain.iter().enumerate().rev().filter(|(_, a)| **a) {
index_map[old_idx] = Some(new_arena.len());
for (old_idx, keep) in retain.iter().enumerate().rev() {
let mut node = self.arena.remove(old_idx);
node.parent = node.parent.and_then(|p| index_map[p]);
node.children = node.children.iter().filter_map(|&c| index_map[c]).collect();
new_arena.push(node);
if *keep {
index_map[old_idx] = Some(new_arena.len());
node.parent = node.parent.and_then(|p| index_map[p]);
node.children = node.children.iter().filter_map(|&c| index_map[c]).collect();
new_arena.push(node);
}
}
self.arena = new_arena;