adventcode21/day13.py at master · mgrank/adventcode21 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
dots = []
folds = []

def printDots(dots):
  w = max(dots, key=lambda d: d[0])[0]
  h = max(dots, key=lambda d: d[1])[1]

  for y in range(h+1):
    for x in range(w+1):
      if (x,y) in dots:
        print('#', sep='', end='')
      else:
        print('.', sep='', end='')
    print()

with open("day13full.txt") as f:
  for line in f:
    line = line.strip()
    if line == '':
      break
    x, y = [int(s) for s in line.split(',')]
    dots += [(x,y)]

  for line in f:
    part = line.split()[2]
    axis, coord = part.split('=')
    folds += [ (axis, int(coord)) ]

print(dots)
print(folds)

for fold, fcoord in folds:
  newDots = []
  for x, y in dots:
    if fold == 'x':
      if x > fcoord:
        x = 2 * fcoord - x
    if fold == 'y':
      if y > fcoord:
        y = 2 * fcoord - y

    newDot = (x, y)
    if newDot not in newDots:
      newDots.append(newDot)
  dots = newDots

print(len(dots))
printDots(dots)