aoc

commit 50b41d675539987dc9402de6f33a545c9732594b
parent ef5c772e61c4fbc8bbb78f75c93706abc4e59438
Author: Sebastiano Tronto <sebastiano@tronto.net>
Date:   Sat,  5 Jul 2025 00:07:13 +0200

Some performance improvements

Diffstat:
M
2022/19/a.rs
 | 
2
+-
M
2022/19/common.rs
 | 
58
+++++++++++++++++++++++++++++++++++-----------------------

2 files changed, 36 insertions(+), 24 deletions(-)
diff --git a/2022/19/a.rs b/2022/19/a.rs
@@ -8,7 +8,7 @@ fn main() {
     let mut i = 1;
     let mut sum = 0;
     for bp in read_blueprints_from_stdin() {
-        let mg = most_geodes(&bp, Status::new(), MINUTES, &mut mem);
+        let mg = most_geodes(&bp, Status::new(), MINUTES, &mut mem, 0);
         println!("{i}: {mg}");
         sum += i * mg;
         mem.clear();
diff --git a/2022/19/common.rs b/2022/19/common.rs
@@ -122,48 +122,60 @@ pub fn most_geodes(
     bp: &Blueprint,
     s: Status,
     m: i32,
-    mem: &mut HashMap<(Status, i32), i32>
+    mem: &mut HashMap<(Status, i32), i32>,
+    lower_bound: i32
 ) -> i32 {
-    if let Some(r) = mem.get(&(s, m)) { return *r; }
     if m == 0 { return s.resources.geo; }
+    if m == 1 { return s.resources.geo + s.bots.geo; }
+
+    // Estimate how much we can get at most, assuming one geode bot
+    // can be built each turn from now on
+    let upper_bound = s.resources.geo + s.bots.geo + (m-1)*(m-2)/2;
+    if upper_bound <= lower_bound { return 0; }
+
+    if let Some(r) = mem.get(&(s, m)) { return *r; }
 
     let mut new_status = s;
     new_status.resources += &new_status.bots;
 
-    let mut result = most_geodes(bp, new_status, m-1, mem);
+    let mut result = 0;
 
-    if m > 4 && s.resources >= bp.ore_bot_cost {
-        new_status.resources -= &bp.ore_bot_cost;
-        new_status.bots.ore += 1;
-        result = max(result, most_geodes(bp, new_status, m-1, mem));
-        new_status.bots.ore -= 1;
-        new_status.resources += &bp.ore_bot_cost;
-    }
+    if m > 1 && s.resources >= bp.geo_bot_cost {
+        new_status.resources -= &bp.geo_bot_cost;
+        new_status.bots.geo += 1;
+        result = most_geodes(bp, new_status, m-1, mem, result);
 
-    if m > 3 && s.resources >= bp.cla_bot_cost {
-        new_status.resources -= &bp.cla_bot_cost;
-        new_status.bots.cla += 1;
-        result = max(result, most_geodes(bp, new_status, m-1, mem));
-        new_status.bots.cla -= 1;
-        new_status.resources += &bp.cla_bot_cost;
+        // If a geode bot can be built, it is always the best thing to do
+        mem.insert((s, m), result);
+        return result;
     }
 
     if m > 2 && s.resources >= bp.obs_bot_cost {
         new_status.resources -= &bp.obs_bot_cost;
         new_status.bots.obs += 1;
-        result = max(result, most_geodes(bp, new_status, m-1, mem));
+        result = max(result, most_geodes(bp, new_status, m-1, mem, result));
         new_status.bots.obs -= 1;
         new_status.resources += &bp.obs_bot_cost;
     }
 
-    if m > 1 && s.resources >= bp.geo_bot_cost {
-        new_status.resources -= &bp.geo_bot_cost;
-        new_status.bots.geo += 1;
-        result = max(result, most_geodes(bp, new_status, m-1, mem));
-        new_status.bots.geo -= 1;
-        new_status.resources += &bp.geo_bot_cost;
+    if m > 3 && s.resources >= bp.cla_bot_cost {
+        new_status.resources -= &bp.cla_bot_cost;
+        new_status.bots.cla += 1;
+        result = max(result, most_geodes(bp, new_status, m-1, mem, result));
+        new_status.bots.cla -= 1;
+        new_status.resources += &bp.cla_bot_cost;
     }
 
+    if m > 4 && s.resources >= bp.ore_bot_cost {
+        new_status.resources -= &bp.ore_bot_cost;
+        new_status.bots.ore += 1;
+        result = max(result, most_geodes(bp, new_status, m-1, mem, result));
+        new_status.bots.ore -= 1;
+        new_status.resources += &bp.ore_bot_cost;
+    }
+
+    result = max(result, most_geodes(bp, new_status, m-1, mem, result));
+
     mem.insert((s, m), result);
     result
 }