algo — C++23 日本語リファレンス GitHub

test/yosupo/math/sqrt_mod.test.cpp

AC · 最終更新 2026-09-07 · GitHub

問題: https://judge.yosupo.jp/problem/sqrt_mod

依存するヘッダ(4)
実装
// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/sqrt_mod
#include <iostream>
#include "number_theory/modint.hpp"
#include "number_theory/root_mod.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 (auto r = sqrt_mod(x)) std::cout << *r << '\n';
        else std::cout << -1 << '\n';
    }

    return 0;
}