115 lines
2.6 KiB
Python
115 lines
2.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# 2021 - Advent Of Code - 13
|
||
|
|
|
||
|
|
|
||
|
|
def parse_file(file):
|
||
|
|
parsed_dots = []
|
||
|
|
parsed_folds = []
|
||
|
|
with open(file) as f:
|
||
|
|
for line in f.readlines():
|
||
|
|
if line.startswith('fold along '):
|
||
|
|
_, f = line.strip().split('fold along ')
|
||
|
|
parsed_folds.append(f.split('='))
|
||
|
|
elif line == '\n':
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
dot = [int(d) for d in line.strip().split(',')]
|
||
|
|
parsed_dots.append(dot)
|
||
|
|
return parsed_dots, parsed_folds
|
||
|
|
|
||
|
|
|
||
|
|
def find_max(dot_list):
|
||
|
|
mx = 0
|
||
|
|
my = 0
|
||
|
|
for d in dot_list:
|
||
|
|
if d[0] > mx:
|
||
|
|
mx = d[0]
|
||
|
|
if d[1] > my:
|
||
|
|
my = d[1]
|
||
|
|
|
||
|
|
return mx, my
|
||
|
|
|
||
|
|
|
||
|
|
def place_dots(g, dots_list):
|
||
|
|
for dot in dots_list:
|
||
|
|
g[dot[1]][dot[0]] = 1
|
||
|
|
|
||
|
|
|
||
|
|
def print_grid(g, fold=-1):
|
||
|
|
for y, _ in enumerate(g):
|
||
|
|
if y == fold:
|
||
|
|
for x, _ in enumerate(g[y]):
|
||
|
|
print('-', end='')
|
||
|
|
else:
|
||
|
|
for x, _ in enumerate(g[y]):
|
||
|
|
if g[y][x] == 1:
|
||
|
|
print('#', end='')
|
||
|
|
else:
|
||
|
|
print('.', end='')
|
||
|
|
print()
|
||
|
|
|
||
|
|
|
||
|
|
def fold_horizontally(g, fold_y):
|
||
|
|
print(f'folding y={fold_y}')
|
||
|
|
for y in range(fold_y, maxy+1):
|
||
|
|
folding_dist = y - fold_y
|
||
|
|
new_y = fold_y - folding_dist
|
||
|
|
|
||
|
|
# place new dots
|
||
|
|
for x, _ in enumerate(g[y]):
|
||
|
|
if g[y][x] == 1:
|
||
|
|
g[new_y][x] = 1
|
||
|
|
|
||
|
|
# cut the other half of the grid
|
||
|
|
for i in range(fold_y+1):
|
||
|
|
g.pop()
|
||
|
|
|
||
|
|
|
||
|
|
def fold_vertically(g, fold_x):
|
||
|
|
print(f'folding x={fold_x}')
|
||
|
|
for y, _ in enumerate(g):
|
||
|
|
# place new dots
|
||
|
|
for x in range(fold_x, maxx+1):
|
||
|
|
folding_dist = x - fold_x
|
||
|
|
new_x = fold_x - folding_dist
|
||
|
|
if g[y][x] == 1:
|
||
|
|
g[y][new_x] = 1
|
||
|
|
|
||
|
|
# cut the other half of the grid
|
||
|
|
g[y] = g[y][:fold_x]
|
||
|
|
|
||
|
|
|
||
|
|
def count_dots(g):
|
||
|
|
count = 0
|
||
|
|
for y, _ in enumerate(g):
|
||
|
|
for x, _ in enumerate(g[y]):
|
||
|
|
if g[y][x] == 1:
|
||
|
|
count += 1
|
||
|
|
|
||
|
|
return count
|
||
|
|
|
||
|
|
|
||
|
|
# dots, folds = parse_file('input_example.txt')
|
||
|
|
dots, folds = parse_file('input.txt')
|
||
|
|
|
||
|
|
maxx, maxy = find_max(dots)
|
||
|
|
grid = [[0] * (maxx+1) for _ in range(maxy+1)]
|
||
|
|
|
||
|
|
# for f in folds:
|
||
|
|
# print(f'folding {f[0]} = {f[1]}')
|
||
|
|
# for d in dots:
|
||
|
|
# print(f'dot: {d}')
|
||
|
|
|
||
|
|
place_dots(grid, dots)
|
||
|
|
# print_grid(grid, 7)
|
||
|
|
print(f'grid size: {maxx}, {maxy}')
|
||
|
|
|
||
|
|
# folding_y = int(folds[0][1])
|
||
|
|
# print()
|
||
|
|
# fold_horizontally(grid, folding_y)
|
||
|
|
# print()
|
||
|
|
fold_vertically(grid, 655)
|
||
|
|
|
||
|
|
# print_grid(grid)
|
||
|
|
print(f'number of dots: {count_dots(grid)}')
|