cp_library/alg/iter/sort/sort_groups_fn.py

• View this file on GitHub
• Last update: 2025-06-20 03:24:59+09:00

Verified with

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

Code

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

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