Sqrt Mod (verify/number_theory/modsqrt.test.cpp)

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

Depends on

• cp-algo/math/affine.hpp
• cp-algo/math/common.hpp
• cp-algo/number_theory/discrete_sqrt.hpp
• cp-algo/number_theory/modint.hpp
• cp-algo/random/rng.hpp

Code

// @brief Sqrt Mod
#define PROBLEM "https://judge.yosupo.jp/problem/sqrt_mod"
#pragma GCC optimize("Ofast,unroll-loops")
#pragma GCC target("tune=native")
#include "cp-algo/number_theory/discrete_sqrt.hpp"
#include <bits/stdc++.h>

using namespace std;
using namespace cp_algo::math;
using base = dynamic_modint<>;

void solve() {
    int y, p;
    cin >> y >> p;
    base::switch_mod(p);
    auto res = sqrt(base(y));
    if(res) {
        cout << *res << "\n";
    } else {
        cout << -1 << "\n";
    }
}

signed main() {
    //freopen("input.txt", "r", stdin);
    ios::sync_with_stdio(0);
    cin.tie(0);
    int t = 1;
    cin >> t;
    while(t--) {
        solve();
    }
}

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj_resolve/resolver.py", line 181, 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 252, in bundle
    bundler.update(path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 327, in update
    assert len(lines) == len(uncommented_lines)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AssertionError

Test cases

Env	Name	Status	Elapsed	Memory
g++	example_00	AC	5 ms	4 MB
g++	max_random_00	AC	131 ms	4 MB
g++	max_random_01	AC	132 ms	4 MB
g++	max_random_02	AC	136 ms	4 MB
g++	max_random_03	AC	138 ms	4 MB
g++	max_random_04	AC	137 ms	3 MB
g++	mod_998244353_00	AC	189 ms	4 MB
g++	mod_998244353_01	AC	190 ms	4 MB
g++	random_00	AC	28 ms	4 MB
g++	random_01	AC	30 ms	4 MB
g++	random_02	AC	87 ms	4 MB
g++	random_03	AC	39 ms	4 MB
g++	random_04	AC	135 ms	4 MB

Back to top page