test/aoj/jag/dsu_on_tree.test.cpp

• View this file on GitHub
• Last update: 2025-06-13 13:22:21+09:00
• Problem: https://onlinejudge.u-aizu.ac.jp/problems/2995

Depends on

• 重み付きグラフ (lib/graph/graph.hpp)
• lib/template/template.hpp
• DSU on Tree (lib/tree/dsu_on_tree.hpp)
• 素集合データ構造 (lib/tree/union_find.hpp)

Code

// competitive-verifier: PROBLEM https://onlinejudge.u-aizu.ac.jp/problems/2995
#include "tree/dsu_on_tree.hpp"
#include <iostream>
#include <unordered_map>
#include <utility>
#include <vector>
#include "graph/graph.hpp"
#include "tree/union_find.hpp"

int main(void) {
    int n, k;
    std::cin >> n >> k;
    Graph<void> g(n);
    g.input_edges(n - 1);
    dsu_on_tree dsu(g);
    std::vector<std::pair<int, int>> a(n);
    for (auto &[x, y] : a) std::cin >> x >> y, --x, --y;
    std::vector<int> ans(n);
    int sum = 0;
    union_find uf(k);
    std::vector<bool> flag(k);
    std::vector<int> history;
    auto rem = [&](int x) { ans[x] = sum; };
    auto clear = [&]() {
        sum = 0;
        for (int x : history) {
            uf[x] = -1;
            flag[x] = false;
        }
        history.clear();
    };
    auto f = [&](int x, int y) {
        if (x == y) {
            if (!flag[x]) {
                ++sum;
                flag[x] = true;
            }
        } else {
            if (!flag[x] || !flag[y]) ++sum;
            flag[x] = flag[x] | flag[y];
        }
    };
    auto query = [&](int v) {
        auto [s, t] = a[v];
        uf.unite(s, t, f);
        history.emplace_back(s);
        history.emplace_back(t);
    };
    dsu.solve(rem, clear, query);
    for (int i = 0; i < n; ++i) std::cout << ans[i] << '\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: tree/dsu_on_tree.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	00_sample_00	AC	2 ms	3 MB
g++	00_sample_01	AC	2 ms	3 MB
g++	00_sample_02	AC	2 ms	4 MB
g++	01_small_01	AC	2 ms	4 MB
g++	01_small_02	AC	2 ms	3 MB
g++	02_mid_01	AC	4 ms	4 MB
g++	02_mid_02	AC	4 ms	4 MB
g++	03_large_01	AC	77 ms	11 MB
g++	03_large_02	AC	109 ms	14 MB
g++	04_max_01	AC	119 ms	15 MB
g++	04_max_02	AC	119 ms	15 MB
g++	05_line_01	AC	130 ms	35 MB
g++	05_line_02	AC	126 ms	25 MB
g++	06_uni_01	AC	93 ms	16 MB
g++	06_uni_02	AC	95 ms	15 MB
g++	07_mukade_01	AC	120 ms	15 MB
g++	07_mukade_02	AC	114 ms	15 MB
g++	08_long_01	AC	138 ms	22 MB
g++	08_long_02	AC	123 ms	23 MB
g++	09_binary_01	AC	125 ms	16 MB
g++	09_binary_02	AC	124 ms	16 MB
g++	10_knary_01	AC	107 ms	16 MB
g++	10_knary_02	AC	103 ms	15 MB
g++	11_one_01	AC	91 ms	16 MB
g++	11_one_02	AC	120 ms	23 MB
g++	11_one_03	AC	124 ms	16 MB
g++	11_one_04	AC	107 ms	16 MB
g++	11_one_05	AC	119 ms	16 MB
g++	11_one_06	AC	135 ms	35 MB
g++	11_one_07	AC	93 ms	16 MB
g++	11_one_08	AC	108 ms	24 MB
g++	11_one_09	AC	118 ms	16 MB
g++	11_one_10	AC	111 ms	16 MB
g++	11_one_11	AC	114 ms	16 MB
g++	11_one_12	AC	118 ms	35 MB
g++	11_one_13	AC	91 ms	16 MB
g++	11_one_14	AC	121 ms	23 MB
g++	11_one_15	AC	126 ms	16 MB
g++	11_one_16	AC	111 ms	16 MB
g++	12_few_01	AC	118 ms	16 MB
g++	12_few_02	AC	130 ms	35 MB
g++	12_few_03	AC	90 ms	16 MB
g++	12_few_04	AC	118 ms	24 MB
g++	12_few_05	AC	127 ms	16 MB
g++	12_few_06	AC	120 ms	16 MB
g++	13_some_01	AC	119 ms	16 MB
g++	13_some_02	AC	131 ms	35 MB
g++	13_some_03	AC	92 ms	16 MB
g++	13_some_04	AC	123 ms	25 MB
g++	13_some_05	AC	125 ms	16 MB
g++	13_some_06	AC	119 ms	16 MB
g++	14_many_01	AC	120 ms	16 MB
g++	14_many_02	AC	131 ms	35 MB
g++	14_many_03	AC	94 ms	16 MB
g++	14_many_04	AC	120 ms	24 MB
g++	14_many_05	AC	129 ms	16 MB
g++	14_many_06	AC	120 ms	16 MB

Back to top page