src/$tests/tree/HLD.test.py

View this file on GitHub
Last update: 2023-07-23 08:45:17+09:00
Problem: https://judge.yosupo.jp/problem/vertex_add_path_sum
Link: View error logs on GitHub Actions

Depends on

Code

# verification-helper: PROBLEM https://judge.yosupo.jp/problem/vertex_add_path_sum
from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parent.parent.parent.parent))
from src.tree.HLD import HLD
from src.data_structure.BIT import BIT


n, Q = map(int, input().split())
A = list(map(int, input().split()))
G = HLD(n)
G.read_edges(0)
G.build()
A = G.reorder(A)
bit = BIT(n)
for i, a in enumerate(A):
    bit.add(i, a)

for _ in range(Q):
    query = list(map(int, input().split()))
    if query[0] == 0:
        p, x = query[1:]
        bit.add(G.L[p], x)
    else:
        u, v = query[1:]
        ans = 0
        for l, r in G.get_path(u, v):
            l = G.L[l]
            r = G.L[r]
            if l > r:
                l, r = r, l
            ans += bit.sum(l, r + 1)
        print(ans)

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 81, in _render_source_code_stat
    bundled_code = language.bundle(
                   ^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/python.py", line 108, in bundle
    raise NotImplementedError
NotImplementedError