diff options
| -rw-r--r-- | 2023/14/solve.py | 31 |
1 files changed, 25 insertions, 6 deletions
diff --git a/2023/14/solve.py b/2023/14/solve.py index cedaf8f..ae8f674 100644 --- a/2023/14/solve.py +++ b/2023/14/solve.py @@ -24,26 +24,45 @@ def tilt(mp, d): return mp -def part1(): +def load(mp): res = 0 - - mp = tilt(L, 0) - for i, l in enumerate(mp): res += (len(mp) - i) * l.count("O") - print(res) + return res + + +def part1(): + res = 0 + mp = tilt(L, 0) + print(load(mp)) def part2(): res = 0 + hist = {} + cycle = 0 + mp = L - for i in range(100): + 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") |