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

remove pointless check, tabify

This commit is contained in:
Luke Hubmayer-Werner 2022-12-12 17:21:51 +10:30
parent 4cea8b4043
commit ba924fc41c
1 changed files with 33 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
2022/day12.py
View File

@ -9,61 +9,42 @@ accszExk
acctuvwj
abdefghi'''.strip().split('\n')
def make_heightmap(lines: list[str]) -> tuple[ArrayLike, ArrayLike, ArrayLike]:
heightmap = np.zeros((len(lines[0]), len(lines)), dtype=dtype)
for y, line in enumerate(lines):
for x, c in enumerate(line):
if c == 'S':
heightmap[x,y] = 0
start = np.array([x,y], dtype=dtype)
elif c == 'E':
heightmap[x,y] = 25
end = np.array([x,y], dtype=dtype)
else:
heightmap[x,y] = ord(c) - ord('a')
return heightmap, start, end
# def make_cell_costs(position, heightmap):
# init_val = 9_999_999
# costs = np.full_like(heightmap, init_val)
# pos_t = tuple(position)
# costs[pos_t] = 0
# curr_positions = {pos_t}
# while (costs == init_val).any() and len(curr_positions) > 0:
# next_positions = set()
# for pos_t in curr_positions:
# n_h_limit = heightmap[pos_t] + 1
# n_cost = costs[pos_t] + 1
# for d in directions_array:
# next_t = (pos_t[0] + d[0], pos_t[1] + d[1])
# if (0 <= next_t[0] < heightmap.shape[0]) and (0 <= next_t[1] < heightmap.shape[1]):
# hn = heightmap[next_t]
# if n_h_limit >= hn and costs[next_t] > n_cost:
# costs[next_t] = n_cost
# next_positions.add(next_t)
# curr_positions = next_positions
# return costs
def make_heightmap(lines: list[str]) -> tuple[ArrayLike, ArrayLike, ArrayLike]:
heightmap = np.zeros((len(lines[0]), len(lines)), dtype=dtype)
for y, line in enumerate(lines):
for x, c in enumerate(line):
if c == 'S':
heightmap[x,y] = 0
start = np.array([x,y], dtype=dtype)
elif c == 'E':
heightmap[x,y] = 25
end = np.array([x,y], dtype=dtype)
else:
heightmap[x,y] = ord(c) - ord('a')
return heightmap, start, end
def make_reversed_cell_costs(position, heightmap):
init_val = 9_999_999
costs = np.full_like(heightmap, init_val)
pos_t = tuple(position)
costs[pos_t] = 0
curr_positions = {pos_t}
while (costs == init_val).any() and len(curr_positions) > 0:
next_positions = set()
for pos_t in curr_positions:
n_h_limit = heightmap[pos_t] - 1
n_cost = costs[pos_t] + 1
for d in directions_array:
next_t = (pos_t[0] + d[0], pos_t[1] + d[1])
if (0 <= next_t[0] < heightmap.shape[0]) and (0 <= next_t[1] < heightmap.shape[1]):
hn = heightmap[next_t]
if n_h_limit <= hn and costs[next_t] > n_cost:
costs[next_t] = n_cost
next_positions.add(next_t)
curr_positions = next_positions
return costs
init_val = 9_999_999
costs = np.full_like(heightmap, init_val)
pos_t = tuple(position)
costs[pos_t] = 0
curr_positions = {pos_t}
while len(curr_positions) > 0:
next_positions = set()
for pos_t in curr_positions:
n_h_limit = heightmap[pos_t] - 1
n_cost = costs[pos_t] + 1
for d in directions_array:
next_t = (pos_t[0] + d[0], pos_t[1] + d[1])
if (0 <= next_t[0] < heightmap.shape[0]) and (0 <= next_t[1] < heightmap.shape[1]):
hn = heightmap[next_t]
if n_h_limit <= hn and costs[next_t] > n_cost:
costs[next_t] = n_cost
next_positions.add(next_t)
curr_positions = next_positions
return costs
# heightmap, start, end = make_heightmap(sample_lines)
heightmap, start, end = make_heightmap(lines)