Finished 2023-12-16 p1

This commit is contained in:
Frederik Baerentsen 2023-12-16 22:13:25 -05:00
parent 141dfa3282
commit c93efcc3ab

168
2023/day16/part1.py Normal file
View File

@ -0,0 +1,168 @@
import sys
import os
from pprint import pprint
import time
import math
#colors
from termcolor import colored
sys.setrecursionlimit(3000)
grid = []
def p(x,*args):
for idx,i in enumerate(x):
for jdx,j in enumerate(i):
if j == '#':
print(colored(j,'red'),end='')
elif len(args) > 0:
if (idx,jdx) == args[1]:
print(colored(j,'red'),end='')
elif (idx,jdx) in args[0]:
print(colored(j,'green'),end='')
else:
print(j,end='')
else:
print(j,end='')
print()
input_f = ''
if len(sys.argv) == 1:
input_f = 'test'
else:
input_f = sys.argv[1]
with open(input_f) as file:
for line in file:
grid.append(list(line.rstrip()))
visited = []
visited.append((0,0))
seen = {}
def travel(grid,l,direction):
global visited
current = (l[0],l[1])
try:
if seen[current] == direction:
return
except:
seen[current] = direction
if len(sys.argv) == 3:
p(grid,visited,current)
print()
time.sleep(float(sys.argv[2]))
r = l[0]
c = l[1]
next_loc = ()
#if r >= len(grid) or c >= len(grid[0]) or r < 0 or c < 0:
# return
#else:
# print('Current: ' + str(current) + ' [' + grid[r][c] + ']', end=' ')
# print('moving ' + direction + ' to ',end='')
if direction == 'r':
#print(str((r,c+1)))
if c+1 < len(grid[0]):
next_loc = (r,c+1)
else:
return
#travel(grid,r,c+1,'r')
if direction == 'l':
#print(str((r,c-1)))
if c-1 >= 0:
next_loc = (r,c-1)
else:
return
#travel(grid,r,c-1,'l')
if direction == 'd':
#print(str((r+1,c)))
if r+1 < len(grid):
next_loc = (r+1,c)
else:
return
#travel(grid,r+1,c,'d')
if direction == 'u':
#print(str((r-1,c)))
if r-1 >= 0:
next_loc = (r-1,c)
else:
return
#travel(grid,r-1,c,'u')
visited.append(next_loc)
match grid[next_loc[0]][next_loc[1]]:
case '.':
travel(grid,next_loc,direction)
case '|':
if direction == 'l' or direction == 'r':
travel(grid,next_loc,'u')
travel(grid,next_loc,'d')
else:
travel(grid,next_loc,direction)
case '\\':
if direction == 'r':
travel(grid,next_loc,'d')
if direction == 'l':
travel(grid,next_loc,'u')
if direction == 'd':
travel(grid,next_loc,'r')
if direction == 'u':
travel(grid,next_loc,'l')
case '-':
if direction == 'u' or direction == 'd':
travel(grid,next_loc,'r')
travel(grid,next_loc,'l')
else:
travel(grid,next_loc,direction)
case '/':
if direction == 'r':
travel(grid,next_loc,'u')
if direction == 'l':
travel(grid,next_loc,'d')
if direction == 'd':
travel(grid,next_loc,'l')
if direction == 'u':
travel(grid,next_loc,'r')
travel(grid,(0,-1),'r')
t_grid = grid
for i in visited:
t_grid[i[0]][i[1]] = '#'
print()
p(t_grid)
count = 0
for idx,i in enumerate(t_grid):
for jdx,j in enumerate(i):
if t_grid[idx][jdx] == '#':
count += 1
print(count)
print(len(visited))
print(len(set(visited)))