cp_library/bit/pack/pack_indices_fn.py

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

Required by

• cp_library/math/invcnt_fn.py

Verified with

• test/atcoder/arc/arc136_b_inversion_cnt_fn.test.py

Code

import cp_library.__header__
import cp_library.bit.__header__
import cp_library.bit.pack.__header__
def pack_indices(A, s): return [a<<s|i for i,a in enumerate(A)]

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


def pack_indices(A, s): return [a<<s|i for i,a in enumerate(A)]

Back to top page