verify/tree/hld/hld_vertex_add_path_sum_test.nim

View this file on GitHub
Last update: 2025-02-07 22:33:05+09:00
Problem: https://judge.yosupo.jp/problem/vertex_add_path_sum

Depends on

Code

# verification-helper: PROBLEM https://judge.yosupo.jp/problem/vertex_add_path_sum
import sequtils, strutils, sugar
import cplib/graph/graph
import cplib/collections/segtree
import cplib/tree/heavylightdecomposition
proc scanf(formatstr: cstring){.header: "<stdio.h>", varargs.}
proc ii(): int {.inline.} = scanf("%lld\n", addr result)

var n, q = ii()
var a = newSeqWith(n, ii())
var g = initUnWeightedUnDirectedStaticGraph(n)
for i in 0..<n-1:
    var u, v = ii()
    g.add_edge(u, v)
g.build

var hld = initHld(g, 0)
var v = (0..<n).toSeq.mapIt(hld.toVtx(it)).mapIt(a[it])

var seg = newSegWith(v, l+r, 0)
var ans = newSeqOfCap[int](q)

for i in 0..<q:
    var t = ii()
    if t == 0:
        var p, x = ii()
        seg[hld.toSeq(p)] = seg[hld.toSeq(p)] + x
    else:
        var u, v = ii()
        var l = hld.lca(u, v)
        var a = 0
        for (l, r) in hld.path(l, u, true, false): a += seg.get(l..<r)
        for (l, r) in hld.path(l, v, false, false): a += seg.get(l..<r)
        ans.add(a)
echo ans.join("\n")

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/onlinejudge_verify/languages/nim.py", line 86, in bundle
    raise NotImplementedError
NotImplementedError