cp_library/alg/iter/argsort_bounded_fn.py

• View this file on GitHub
• Last update: 2025-04-03 08:59:41+09:00

Verified with

• test/library-checker/graph/incremental_scc.test.py

Code

import cp_library.alg.iter.__header__

def argsort_bounded(A, mx):
    I, cnt, t = [0]*len(A), [0]*(mx+1), 0
    for a in A: cnt[a] += 1
    for i in range(mx+1): cnt[i], t = t, t+cnt[i]
    for i,a in enumerate(A): I[cnt[a]] = i; cnt[a] += 1
    return I

'''
╺━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╸
             https://kobejean.github.io/cp-library               
'''

def argsort_bounded(A, mx):
    I, cnt, t = [0]*len(A), [0]*(mx+1), 0
    for a in A: cnt[a] += 1
    for i in range(mx+1): cnt[i], t = t, t+cnt[i]
    for i,a in enumerate(A): I[cnt[a]] = i; cnt[a] += 1
    return I

Back to top page