nova/aoc2022
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

some minor pruning, but whatever

Browse Source
This commit is contained in:
Jur van den Berg
2022-12-19 23:43:04 +01:00
parent dee8ebe78e
commit 7c6b17c3e2
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/day19.rs
View File

@@ -113,32 +113,35 @@ fn find_best_geodes(bp: &Blueprint, time: usize) -> usize {
obsidian: state.obsidian - bp.geode_bot_cost.1 + state.obsidian_bots,
..state.tick()
};
if visited.insert(next.clone()) {
if visited.insert(next) {
stack.push(next);
}
}
// Obsidian bots are nice too
if state.ore >= bp.obsidian_bot_cost.0 && state.clay >= bp.obsidian_bot_cost.1 {
// Obsidian bots are nice too, unless we have enough
if state.ore >= bp.obsidian_bot_cost.0
&& state.clay >= bp.obsidian_bot_cost.1
&& state.obsidian_bots < bp.geode_bot_cost.1
{
let next = State {
obsidian_bots: state.obsidian_bots + 1,
ore: state.ore - bp.obsidian_bot_cost.0 + state.ore_bots,
clay: state.clay - bp.obsidian_bot_cost.1 + state.clay_bots,
..state.tick()
};
if visited.insert(next.clone()) {
if visited.insert(next) {
stack.push(next);
}
}
// Clay?
if state.ore >= bp.clay_bot_cost {
// Clay? also prune if we have enough for obsidian bots
if state.ore >= bp.clay_bot_cost && state.clay_bots < bp.obsidian_bot_cost.1 {
let next = State {
clay_bots: state.clay_bots + 1,
ore: state.ore - bp.clay_bot_cost + state.ore_bots,
..state.tick()
};
if visited.insert(next.clone()) {
if visited.insert(next) {
stack.push(next);
}
}
@@ -150,14 +153,14 @@ fn find_best_geodes(bp: &Blueprint, time: usize) -> usize {
ore: state.ore - bp.ore_bot_cost + state.ore_bots,
..state.tick()
};
if visited.insert(next.clone()) {
if visited.insert(next) {
stack.push(next);
}
}
// Final option, do nothing
let next = state.tick();
if visited.insert(next.clone()) {
if visited.insert(next) {
stack.push(next);
}
}