-
Notifications
You must be signed in to change notification settings - Fork 0
/
day14.py
76 lines (62 loc) · 1.48 KB
/
day14.py
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
"""
Day 14: Parabolic Reflector Dish
"""
from itertools import islice
SAMPLE_INPUT = """
O....#....
O.OO#....#
.....##...
OO.#O....O
.O.....O#.
O.#..O.#.#
..O..#O..O
.......O..
#....###..
#OO..#....
"""
def _tilt(data):
data = list(map(list, data))
for x in range(max(map(len, data))):
y0 = 0
while y0 < len(data):
n, y1 = 0, y0
while y1 < len(data):
c = data[y1][x]
if c == "O":
n += 1
elif c == "#":
break
y1 += 1
for y in range(y0, y1):
data[y][x] = "O" if y < y0 + n else "."
y0 = y1 + 1
return data
def _spin(data):
for _ in range(3):
data = list(zip(*_tilt(data)[::-1]))
return tuple(map("".join, zip(*_tilt(data)[::-1])))
def _load(data):
return sum((len(line) - i) * line.count("O") for i, line in enumerate(data))
def part1(data):
"""
>>> part1(SAMPLE_INPUT)
136
"""
return _load(_tilt(filter(None, data.splitlines())))
def part2(data):
"""
>>> part2(SAMPLE_INPUT)
64
"""
n = 1000000000
data = tuple(filter(None, data.splitlines()))
cache = {data: 0}
for i in range(1, n + 1):
data = _spin(data)
if data in cache:
j = cache[data]
data = next(islice(cache.keys(), j + (n - i) % (i - j), None))
break
cache[data] = i
return _load(data)
parts = (part1, part2)