test/yosupo/tree/vertex_set_path_composite.test.cpp

• View this file on GitHub
• Last update: 2026-04-03 17:50:18+09:00
• Problem: https://judge.yosupo.jp/problem/vertex_set_path_composite

Depends on

• Montgomery modular multiplication (lib/internal/internal_math.hpp)
• lib/math/modint.hpp
• lib/segtree/monoid.hpp
• セグメント木 (lib/segtree/segment_tree.hpp)
• Segment Tree on Tree (lib/tree/segment_tree_on_tree.hpp)

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/vertex_set_path_composite
#include <iostream>
#include <vector>
#include "math/modint.hpp"
#include "segtree/monoid.hpp"
#include "tree/segment_tree_on_tree.hpp"

using Mint = modint998;
using M = Affine<Mint>;

int main(void) {
    int n, q;
    std::cin >> n >> q;
    segment_tree_on_tree<M> st(n);
    std::vector<std::pair<Mint, Mint>> p(n);
    for (int i = 0; i < n; ++i) std::cin >> p[i].first >> p[i].second;
    for (int i = 0; i < n - 1; ++i) {
        int u, v;
        std::cin >> u >> v;
        st.add_edge(u, v);
    }
    st.build(p);
    while (q--) {
        int c;
        std::cin >> c;
        if (c == 0) {
            int p, a, b;
            std::cin >> p >> a >> b;
            st.set_vertex(p, {a, b});
        } else {
            int u, v, x;
            std::cin >> u >> v >> x;
            auto ans = st.prod(u, v);
            std::cout << (ans.first * x + ans.second) << '\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: math/modint.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	almost_line_00	AC	784 ms	57 MB
g++	almost_line_01	AC	895 ms	57 MB
g++	example_00	AC	2 ms	4 MB
g++	example_01	AC	2 ms	4 MB
g++	line_00	AC	943 ms	57 MB
g++	line_01	AC	763 ms	57 MB
g++	long-path-decomposition_killer_00	AC	675 ms	57 MB
g++	max_random_00	AC	761 ms	57 MB
g++	max_random_01	AC	671 ms	57 MB
g++	max_random_02	AC	731 ms	57 MB
g++	random_00	AC	709 ms	34 MB
g++	random_01	AC	486 ms	50 MB
g++	random_02	AC	285 ms	17 MB
g++	small_00	AC	5 ms	4 MB
g++	small_01	AC	4 ms	4 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
g++	worst_for_path_decomposition_00	AC	1336 ms	57 MB
g++	worst_for_path_decomposition_01	AC	955 ms	57 MB

Back to top page