40 lines
959 B
Python
40 lines
959 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from typing import Union
|
||
|
|
|
||
|
|
Packet = Union[int, list["Packet"]]
|
||
|
|
|
||
|
|
|
||
|
|
def cmp_packet(lhs: Packet, rhs: Packet) -> int:
|
||
|
|
if isinstance(lhs, int) and isinstance(rhs, int):
|
||
|
|
return lhs - rhs
|
||
|
|
if isinstance(lhs, int):
|
||
|
|
lhs = [lhs]
|
||
|
|
if isinstance(rhs, int):
|
||
|
|
rhs = [rhs]
|
||
|
|
non_equal_cmp = (res for res in map(cmp_packet, lhs, rhs) if res != 0)
|
||
|
|
len_cmp = cmp_packet(len(lhs), len(rhs))
|
||
|
|
return next(non_equal_cmp, len_cmp)
|
||
|
|
|
||
|
|
|
||
|
|
def solve(input: list[str]) -> int:
|
||
|
|
def to_packets(input: str) -> tuple[Packet, Packet]:
|
||
|
|
first, second = input.splitlines()
|
||
|
|
return eval(first), eval(second) # Secret best way to parse JSON
|
||
|
|
|
||
|
|
return sum(
|
||
|
|
i
|
||
|
|
for i, packets in enumerate(input, start=1)
|
||
|
|
if cmp_packet(*to_packets(packets)) <= 0
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
input = sys.stdin.read().split("\n\n")
|
||
|
|
print(solve(input))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|