test/yukicoder/0901.test.cpp

• View this file on GitHub
• Last update: 2026-03-15 03:14:58+09:00
• Problem: https://yukicoder.me/problems/no/901

Depends on

• 線形 Sparse Table (lib/data_structure/linear_sparse_table.hpp)
• スパーステーブル (lib/data_structure/sparse_table.hpp)
• 重み付きグラフ (lib/graph/graph.hpp)
• lib/internal/internal_csr.hpp
• lib/segtree/monoid.hpp
• セグメント木 (lib/segtree/segment_tree.hpp)
• lib/tree/auxiliary_tree.hpp
• オイラーツアー (lib/tree/euler_tour.hpp)
• HL分解 (lib/tree/hld.hpp)
• lib/tree/linear_lca.hpp

Code

// competitive-verifier: PROBLEM https://yukicoder.me/problems/no/901
#include <cstdint>
#include <iostream>
#include <vector>
#include "graph/graph.hpp"
#include "segtree/segment_tree.hpp"
#include "tree/auxiliary_tree.hpp"
#include "tree/hld.hpp"

int main(void) {
    int n;
    std::cin >> n;
    Graph<std::int64_t> g(n);
    g.input_edges(n - 1, 0);
    heavy_light_decomposition hld(g);
    segment_tree<Add<std::int64_t>> st(n);
    auto f = [&](auto self, int x, int p) -> void {
        for (auto e : g[x]) {
            if (e.to() == p) continue;
            st.set(hld.get(e.to()), e.weight());
            self(self, e.to(), x);
        }
    };
    f(f, 0, -1);

    auxiliary_tree_builder at(g);
    int q;
    std::cin >> q;
    while (q--) {
        int k;
        std::cin >> k;
        std::vector<int> x(k);
        for (auto &e : x) std::cin >> e;
        auto tr = at.build(x);
        std::int64_t ans = 0;
        auto g = [&](int x, int y) { ans += st.prod(x, y); };
        auto dfs = [&](auto self, int v, int p) -> void {
            for (auto e : tr[v]) {
                if (e.to() == p) continue;
                hld.for_each_edge(tr.vertex(v), tr.vertex(e.to()), g);
                self(self, e.to(), v);
            }
        };
        dfs(dfs, 0, -1);
        std::cout << ans << '\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: graph/graph.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	challenge01	AC	126 ms	23 MB
g++	sample01	AC	2 ms	4 MB
g++	test_1_01	AC	4 ms	4 MB
g++	test_1_02	AC	4 ms	4 MB
g++	test_1_03	AC	4 ms	4 MB
g++	test_1_04	AC	4 ms	4 MB
g++	test_1_05	AC	4 ms	4 MB
g++	test_2_01	AC	191 ms	20 MB
g++	test_2_02	AC	220 ms	20 MB
g++	test_2_03	AC	186 ms	20 MB
g++	test_2_04	AC	212 ms	20 MB
g++	test_2_05	AC	186 ms	20 MB
g++	test_3_01	AC	183 ms	20 MB
g++	test_3_02	AC	178 ms	20 MB
g++	test_3_03	AC	179 ms	20 MB
g++	test_3_04	AC	166 ms	20 MB
g++	test_3_05	AC	182 ms	20 MB
g++	test_4_01	AC	252 ms	20 MB
g++	test_4_02	AC	265 ms	20 MB
g++	test_4_03	AC	265 ms	20 MB
g++	test_4_04	AC	250 ms	20 MB
g++	test_4_05	AC	242 ms	20 MB
g++	test_5_01	AC	156 ms	21 MB
g++	test_5_02	AC	160 ms	21 MB
g++	test_5_03	AC	156 ms	21 MB
g++	test_5_04	AC	156 ms	21 MB
g++	test_5_05	AC	157 ms	21 MB
g++	test_6_01	AC	307 ms	20 MB
g++	test_6_02	AC	292 ms	20 MB
g++	test_6_03	AC	296 ms	20 MB

Back to top page