test/yosupo/data_structure/unionfind_with_potential.test.cpp

• View this file on GitHub
• Last update: 2025-05-21 23:13:24+09:00
• Problem: https://judge.yosupo.jp/problem/unionfind_with_potential

Depends on

• Montgomery modular multiplication (lib/internal/internal_math.hpp)
• lib/math/modint.hpp
• 重み付き素集合データ構造 (lib/tree/weighted_union_find.hpp)

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/unionfind_with_potential
#include <iostream>
#include <vector>
#include "math/modint.hpp"
#include "tree/weighted_union_find.hpp"

using Mint = modint998;

int main(void) {
    int n, q;
    std::cin >> n >> q;
    weighted_union_find<Mint> uf(n);
    while (q--) {
        int t;
        std::cin >> t;
        if (t == 0) {
            int u, v;
            Mint x;
            std::cin >> u >> v >> x;
            if (uf.same(u, v)) std::cout << (uf.diff(v, u) == x) << '\n';
            else std::cout << uf.unite(v, u, x) << '\n';
        } else {
            int u, v;
            std::cin >> u >> v;
            if (uf.same(u, v)) std::cout << uf.diff(v, u) << '\n';
            else std::cout << -1 << '\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++	example_00	AC	3 ms	4 MB
g++	max_random_00	AC	404 ms	4 MB
g++	max_random_01	AC	417 ms	4 MB
g++	max_random_02	AC	397 ms	4 MB
g++	path_00	AC	434 ms	5 MB
g++	path_01	AC	422 ms	5 MB
g++	path_02	AC	484 ms	5 MB
g++	path_03	AC	457 ms	5 MB
g++	random_00	AC	327 ms	4 MB
g++	random_01	AC	327 ms	4 MB
g++	random_02	AC	286 ms	4 MB
g++	random_03	AC	58 ms	4 MB
g++	random_04	AC	205 ms	4 MB
g++	random_05	AC	279 ms	4 MB
g++	random_06	AC	220 ms	5 MB
g++	random_07	AC	41 ms	4 MB
g++	random_08	AC	98 ms	4 MB
g++	random_09	AC	427 ms	4 MB

Back to top page