summaryrefslogtreecommitdiff
path: root/2024/05/python/main.py
blob: e3a10e249b4219d55c2d46f690ef005784be0f31 (plain)
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
from fileinput import input
from functools import cmp_to_key
from itertools import takewhile

inp = map(str.strip, input())
ordering = set(takewhile(bool, inp))
us = list(inp)


def cmp(x, y):
    return (f"{y}|{x}" in ordering) * 2 - 1  # hehe


silver = 0
gold = 0

for u in us:
    pre = u.split(",")
    post = list(sorted(pre, key=cmp_to_key(cmp)))

    mid = int(post[len(post) // 2])
    if pre == post:
        silver += mid
    else:
        gold += mid

print("silver:", silver)
print("gold:", gold)