NDOJ/judge/utils/ranker.py
2022-05-14 12:57:27 -05:00

14 lines
311 B
Python

from operator import attrgetter
def ranker(iterable, key=attrgetter("points"), rank=0):
delta = 1
last = None
for item in iterable:
new = key(item)
if new != last:
rank += delta
delta = 0
delta += 1
yield rank, item
last = key(item)