sort

src/main.rs

@@ -280,6 +280,13 @@ impl<const TOWER_HEIGHT: usize> Game<TOWER_HEIGHT> {
                     continue;
                 }
 
+                let rings = self.towers[from_tower].rings;
+                if rings[1..].iter().all(|i| i == &rings[0])
+                    && self.towers[to_tower].rings[0].is_none()
+                {
+                    continue;
+                }
+
                 let m = Move::new(from_tower, to_tower);
                 if self.try_make_move(m, true).is_ok() {
                     res.push(m);
@@ -312,10 +319,45 @@ impl<const TOWER_HEIGHT: usize> Game<TOWER_HEIGHT> {
             }
         }
 
-        let hash_state = |s: &[Tower<TOWER_HEIGHT>]| {
+        fn tower_solved<const TOWER_HEIGHT: usize>(tower: &Tower<TOWER_HEIGHT>) -> bool {
+            tower
+                .rings
+                .iter()
+                .all(|x| x.is_some() && x == &tower.rings[0])
+        }
+
+        let hash_state = |towers: &[Tower<TOWER_HEIGHT>]| {
             let mut hasher = DefaultHasher::default();
 
-            for x in s.iter().filter(|i| i.height() != 0).enumerate() {
+            let num_finished_towers = towers.iter().filter(|i| tower_solved(*i)).count();
+
+            num_finished_towers.hash(&mut hasher);
+            let mut filtered_towers = towers
+                .iter()
+                .filter(|i| !tower_solved(i) && i.height() != 0)
+                .enumerate()
+                .collect::<Vec<_>>();
+
+            filtered_towers.sort_by_cached_key(|(index, tower)| {
+                let mut rings = [0; TOWER_HEIGHT];
+
+                let mut count = 0;
+                for (idx, ring) in tower.rings.iter().copied().enumerate() {
+                    let Some(ring) = ring else { break };
+
+                    if rings[idx] == 0 {
+                        count += 1;
+                        for (j, ring2) in tower.rings[idx..].iter().copied().enumerate() {
+                            if ring2 == Some(ring) {
+                                rings[j] = count;
+                            }
+                        }
+                    }
+                }
+                (rings, *index)
+            });
+
+            for x in filtered_towers.into_iter().map(|(_, x)| x).enumerate() {
                 x.hash(&mut hasher);
             }
 
@@ -762,7 +804,7 @@ fn main() -> io::Result<()> {
 
     let mut g = GameGenerator {
         num_extra_towers: 2,
-        num_ring_types: 13,
+        num_ring_types: 16,
     }
     .highest_burried_score::<4>(&seed);