blob: ae8f67417c5e41bc162b8a7d22ccc4920060586e (
plain) (
blame)
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
|
L = [l.strip() for l in open("input").readlines()]
def tilt(mp, d):
transposed = d == 0 or d == 2
if transposed: # north or south
mp = list(map(list, zip(*mp)))
pat = ".O" if d == 0 or d == 1 else "O."
done = False
while not done:
done = True
for l in mp:
for i in range(len(l) - 1):
if f"{l[i]}{l[i+1]}" != pat:
continue
l[i + 1], l[i] = l[i], l[i + 1]
done = False
if transposed: # north or south
mp = list(map(list, zip(*mp)))
return mp
def load(mp):
res = 0
for i, l in enumerate(mp):
res += (len(mp) - i) * l.count("O")
return res
def part1():
res = 0
mp = tilt(L, 0)
print(load(mp))
def part2():
res = 0
hist = {}
cycle = 0
mp = L
i = 0
while True:
mp = tilt(mp, 0)
mp = tilt(mp, 1)
mp = tilt(mp, 2)
mp = tilt(mp, 3)
if cycle:
if i % cycle == 0:
break
else:
key = tuple(tuple(l) for l in mp)
if key in hist:
cycle = i - hist[key] - 1
else:
hist[key] = i
i += 1
res = 0
for j, l in enumerate(mp):
res += (len(mp) - j) * l.count("O")
print(res)
part1()
part2()
|
