40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
# SPDX-License-Identifier: MIT
|
||
|
|
# Copyright (c) 2022 Akumatic
|
||
|
|
#
|
||
|
|
# https://adventofcode.com/2022/day/13
|
||
|
|
|
||
|
|
from json import loads
|
||
|
|
from functools import cmp_to_key
|
||
|
|
|
||
|
|
def read_file(filename: str = "input.txt") -> dict:
|
||
|
|
with open(f"{__file__.rstrip('code.py')}{filename}", "r") as f:
|
||
|
|
return [loads(line.strip()) for line in f.readlines() if line != "\n"]
|
||
|
|
|
||
|
|
def cmp(a, b) -> int:
|
||
|
|
match a, b:
|
||
|
|
case int(), int():
|
||
|
|
return 0 if a == b else (b - a) // abs(b - a)
|
||
|
|
case int(), list():
|
||
|
|
return cmp([a], b)
|
||
|
|
case list(), int():
|
||
|
|
return cmp(a, [b])
|
||
|
|
case list(), list():
|
||
|
|
for pair in zip(a, b):
|
||
|
|
tmp = cmp(*pair)
|
||
|
|
if tmp:
|
||
|
|
return tmp
|
||
|
|
return 0 if len(a) == len(b) else (len(b) - len(a)) // abs(len(b) - len(a))
|
||
|
|
|
||
|
|
def part1(vals: list) -> int:
|
||
|
|
pairs = [vals[i:i+2] for i in range(0, len(vals), 2)]
|
||
|
|
return sum(idx for idx, pair in enumerate(pairs, 1) if cmp(*pair) == 1)
|
||
|
|
|
||
|
|
def part2(vals: list) -> int:
|
||
|
|
ordered = sorted(vals + [[[2]], [[6]]], reverse=True, key=cmp_to_key(cmp))
|
||
|
|
return (ordered.index([[2]]) + 1) * (ordered.index([[6]]) + 1)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
vals = read_file()
|
||
|
|
print(f"Part 1: {part1(vals)}")
|
||
|
|
print(f"Part 2: {part2(vals)}")
|