cp_library/alg/iter/sort_buckets_fn.py

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

Code

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

def sort_buckets(A, N, key=0):
    if isinstance(key,int):
        key = itemgetter(key)
    A.sort(key=key)
    B = [[] for _ in range(N)]
    for k,g in groupby(A, key=key):
        B[k] = list(g)
    return B
    

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

def sort_buckets(A, N, key=0):
    if isinstance(key,int):
        key = itemgetter(key)
    A.sort(key=key)
    B = [[] for _ in range(N)]
    for k,g in groupby(A, key=key):
        B[k] = list(g)
    return B
    

Back to top page