test/yosupo/tree/vertex_add_subtree_sum.test.cpp

• View this file on GitHub
• Last update: 2026-03-15 03:14:58+09:00
• Problem: https://judge.yosupo.jp/problem/vertex_add_subtree_sum

Depends on

• 重み付きグラフ (lib/graph/graph.hpp)
• lib/segtree/monoid.hpp
• セグメント木 (lib/segtree/segment_tree.hpp)
• オイラーツアー (lib/tree/euler_tour.hpp)

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/vertex_add_subtree_sum
#include <cstdint>
#include <iostream>
#include <vector>
#include "segtree/segment_tree.hpp"
#include "tree/euler_tour.hpp"

int main(void) {
    int n, q;
    std::cin >> n >> q;
    std::vector<int> a(n), p(n - 1);
    for (auto &e : a) std::cin >> e;
    for (auto &e : p) std::cin >> e;
    Graph<void> g(n);
    for (int i = 0; i < n - 1; ++i) g.add_edges(i + 1, p[i]);
    euler_tour et(g);
    segment_tree<Add<std::int64_t>> st(n);
    for (int i = 0; i < n; ++i) st.set(et.left(i), a[i]);

    while (q--) {
        int c;
        std::cin >> c;
        if (c == 0) {
            int v, x;
            std::cin >> v >> x;
            st.set(et.left(v), st.get(et.left(v)) + x);
        } else {
            int v;
            std::cin >> v;
            auto [l, r] = et[v];
            std::cout << st.prod(l, r) << '\n';
        }
    }

    return 0;
}

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/resolver.py", line 291, in resolve
    bundled_code = language.bundle(path, basedir=basedir)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus.py", line 242, in bundle
    bundler.update(path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 479, in update
    self._resolve(pathlib.Path(included), included_from=path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 286, in _resolve
    raise BundleErrorAt(path, -1, "no such header")
competitive_verifier.oj.verify.languages.cplusplus_bundle.BundleErrorAt: segtree/segment_tree.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	example_00	AC	3 ms	3 MB
g++	line_00	AC	1089 ms	55 MB
g++	line_01	AC	1038 ms	55 MB
g++	max_random_00	AC	1233 ms	55 MB
g++	max_random_01	AC	1164 ms	55 MB
g++	max_random_02	AC	1149 ms	56 MB
g++	random_00	AC	883 ms	46 MB
g++	random_01	AC	970 ms	52 MB
g++	random_02	AC	495 ms	9 MB
g++	random_03	AC	414 ms	49 MB
g++	random_04	AC	347 ms	36 MB
g++	small_00	AC	6 ms	4 MB
g++	small_01	AC	4 ms	3 MB
g++	small_02	AC	4 ms	4 MB
g++	small_03	AC	6 ms	4 MB
g++	small_04	AC	4 ms	4 MB

Back to top page