test/yosupo/data_structure/undo_union_find.test.cpp

View this file on GitHub
Last update: 2025-06-10 13:56:45+09:00
Problem: https://judge.yosupo.jp/problem/persistent_unionfind

Depends on

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/persistent_unionfind
#include "tree/undo_union_find.hpp"
#include <iostream>
#include <utility>
#include <vector>
#include "graph/graph.hpp"

struct S {
    int k, u, v;
};

int main(void) {
    int n, q;
    std::cin >> n >> q;
    Graph<std::pair<int, int>> g(q + 1);
    std::vector<std::vector<S>> querys(q + 1);
    for (int i = 0; i < q; ++i) {
        int t, k, u, v;
        std::cin >> t >> k >> u >> v;
        if (t == 0) g.add_edges(k + 1, i + 1, {u, v});
        else querys[k + 1].emplace_back(S{i + 1, u, v});
    }

    undo_union_find uf(n);
    std::vector<int> ans(q + 1, -1);
    auto dfs = [&](auto self, int idx, int par) -> void {
        for (auto query : querys[idx]) ans[query.k] = uf.same(query.u, query.v);
        for (auto e : g[idx]) {
            if (e.to() == par) continue;
            bool flag = uf.unite(e.weight().first, e.weight().second);
            self(self, e.to(), idx);
            if (flag) uf.undo();
        }
    };
    dfs(dfs, 0, -1);
    for (int i = 0; i < q + 1; ++i) {
        if (ans[i] != -1) 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/undo_union_find.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	conchon_filliatre_tle_00	AC	165 ms	44 MB
g++	conchon_filliatre_tle_01	AC	164 ms	44 MB
g++	example_00	AC	2 ms	3 MB
g++	hand_00	AC	2 ms	3 MB
g++	max_random_00	AC	188 ms	44 MB
g++	max_random_01	AC	190 ms	44 MB
g++	max_random_02	AC	191 ms	44 MB
g++	max_random_03	AC	192 ms	44 MB
g++	max_random_04	AC	189 ms	44 MB
g++	random_00	AC	140 ms	34 MB
g++	random_01	AC	141 ms	34 MB
g++	random_02	AC	112 ms	28 MB
g++	random_03	AC	27 ms	9 MB
g++	random_04	AC	89 ms	23 MB