53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
|
data = open("input").read().splitlines(False)
|
||
|
|
||
|
import heapq
|
||
|
import string
|
||
|
chars = 'ES'+string.ascii_lowercase
|
||
|
|
||
|
data = [list(map(chars.index, line)) for line in data]
|
||
|
#print(data)
|
||
|
def get(x, y):
|
||
|
return data[y][x]
|
||
|
|
||
|
q = []
|
||
|
for x in range(len(data[0])):
|
||
|
for y in range(len(data)):
|
||
|
if get(x,y) == 1:
|
||
|
data[y][x] = chars.index('a')
|
||
|
start = (x,y)
|
||
|
elif get(x,y) == 0:
|
||
|
data[y][x] = chars.index('z')
|
||
|
goal = (x,y)
|
||
|
elif get(x,y) == 2:
|
||
|
# part 2
|
||
|
q.append((0,(x,y)))
|
||
|
pass
|
||
|
|
||
|
q.append((0, start))
|
||
|
path = {}
|
||
|
while q:
|
||
|
cost, (x,y) = heapq.heappop(q)
|
||
|
if (x,y) in path:
|
||
|
continue
|
||
|
#print(x,y,cost)
|
||
|
if (x,y) == goal:
|
||
|
print('found', cost)
|
||
|
break
|
||
|
here = get(x,y)
|
||
|
path[(x,y)] = cost
|
||
|
|
||
|
n = []
|
||
|
def check(x,y):
|
||
|
if 0 <= y < len(data):
|
||
|
if 0 <= x < len(data[y]):
|
||
|
if get(x,y) <= here+1 and (x,y) not in path:
|
||
|
n.append((cost+1, (x,y)))
|
||
|
check(x-1, y)
|
||
|
check(x+1,y)
|
||
|
check(x, y-1)
|
||
|
check(x, y+1)
|
||
|
|
||
|
for p in n:
|
||
|
heapq.heappush(q, p)
|
||
|
|