cp_library/alg/iter/sort_groups_fn.py

• View this file on GitHub
• Last update: 2025-03-30 20:17:47+09:00

Verified with

• test/atcoder/abc/abc203_e_sort_groups.test.py

Code

import cp_library.alg.iter.__header__
from itertools import groupby
from operator import itemgetter

def sort_groups(A, key=0):
    if isinstance(key,int):
        key = itemgetter(key)
    A.sort(key=key)
    return sorted((k,list(g)) for k,g in groupby(A, key=key))

    

'''
╺━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╸
             https://kobejean.github.io/cp-library               
'''
from itertools import groupby
from operator import itemgetter

def sort_groups(A, key=0):
    if isinstance(key,int):
        key = itemgetter(key)
    A.sort(key=key)
    return sorted((k,list(g)) for k,g in groupby(A, key=key))

    

Back to top page