test/yosupo/math/sqrt_mod.test.cpp

• View this file on GitHub
• Last update: 2025-05-22 02:52:27+09:00
• Problem: https://judge.yosupo.jp/problem/sqrt_mod

Depends on

• Montgomery modular multiplication (lib/internal/internal_math.hpp)
• lib/math/modint.hpp
• lib/math/sqrt.hpp

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/sqrt_mod
#include <iostream>
#include "math/modint.hpp"
#include "math/sqrt.hpp"

int main(void) {
    int t;
    std::cin >> t;
    while (t--) {
        int y, p;
        std::cin >> y >> p;
        modint::set_mod(p);
        modint x(y);
        if (has_sqrt_mod(x)) std::cout << sqrt_mod(x) << '\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	2 ms	3 MB
g++	max_random_00	AC	210 ms	4 MB
g++	max_random_01	AC	204 ms	3 MB
g++	max_random_02	AC	204 ms	4 MB
g++	max_random_03	AC	231 ms	4 MB
g++	max_random_04	AC	204 ms	4 MB
g++	mod_998244353_00	AC	270 ms	3 MB
g++	mod_998244353_01	AC	282 ms	3 MB
g++	random_00	AC	40 ms	4 MB
g++	random_01	AC	41 ms	3 MB
g++	random_02	AC	155 ms	4 MB
g++	random_03	AC	62 ms	4 MB
g++	random_04	AC	201 ms	4 MB

Back to top page