Tree structure that allows efficient both partial sum computation and element updating Fennic tree - Wikipedia - In anthology it’s called Binary lndexed Tree.

No Binary Indexed Tree

  • Kazuhiro Hosaka (Department of Mathematics, The University of Tokyo)

  • 13th JOI Spring Training Camp

  • 2014/03/19

Can be used as a data structure for fast retrieval of the kth value

python

N = 1000
bit = [0] * (N + 1)  # 1-origin
 
def bit_add(pos, val):
    x = pos
    while x <= N:
        bit[x] += val
        x += x & -x  # (x & -x) = rightmost 1 = block width


def bit_sum(pos):
    ret = 0
    x = pos
    while x > 0:
        ret += bit[x]
        x -= x & -x
    return ret


def bit_bisect(lower):
    "find a s.t. v1 + v2 + ... + va >= lower"
    x = 0
    k = 1 << (N.bit_length() - 1)  # largest 2^m <= N
    while k > 0:
        if (x + k <= N and bit[x + k] < lower):
            lower -= bit[x + k]
            x += k
        k //= 2
    return x + 1


bit_add(12, 1)
bit_add(34, 1)
bit_add(56, 1)
print(bit_sum(20))  # => 1
print(bit_sum(40))  # => 2
print(bit_sum(60))  # => 3
print(bit_bisect(2))  # => 34

https://judge.yosupo.jp/submission/12646


This page is auto-translated from /nishio/フェニック木 using DeepL. If you looks something interesting but the auto-translated English is not good enough to understand it, feel free to let me know at @nishio_en. I’m very happy to spread my thought to non-Japanese readers.