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)
|