birdulon
/
AdventOfCode
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

numpy-ify part 2

This commit is contained in:
Luke Hubmayer-Werner 2022-12-12 17:16:15 +10:30
parent 8919f4caae
commit 4cea8b4043
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
2022/day12.py
View File

@ -69,14 +69,5 @@ def make_reversed_cell_costs(position, heightmap):
heightmap, start, end = make_heightmap(lines) heightmap, start, end = make_heightmap(lines)
rev_costs = make_reversed_cell_costs(end, heightmap) rev_costs = make_reversed_cell_costs(end, heightmap)
start_score = rev_costs[*start] print(f'Part 1: {rev_costs[*start]}')
print(f'Part 1: {start_score}') print(f'Part 2: {rev_costs[heightmap == 0].min()}')
start_candidates = {tuple(pos) for pos in np.transpose(np.nonzero(heightmap == 0))}
best_candidate = start
best_score = start_score
for candidate in start_candidates:
score = rev_costs[*candidate]
if score < best_score:
best_candidate = candidate
best_score = score
print(f'Part 2: start at {best_candidate}: cost of {best_score}')