URAL 1132 Square Root（二次剩余定理）题解

题意：

求\(x^2 \equiv a \mod p\) 的所有整数解

思路：

二次剩余定理求解。

参考：

二次剩余Cipolla's algorithm学习笔记

板子：

//二次剩余，p是奇质数
ll ppow(ll a, ll b, ll mod){
    ll ret = 1;
    a = a % mod;
    while(b){
        if(b & 1) ret = ret * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ret;
}
struct TT{
    ll p, d;
};
ll w;
TT mul_er(TT a, TT b, ll mod){
    TT ans;
    ans.p = (a.p * b.p % mod + a.d * b.d % mod * w % mod) % mod;
    ans.d = (a.p * b.d % mod + a.d * b.p % mod) % mod;
    return ans;
}
TT power(TT a, ll b, ll mod){
    TT ret;
    ret.p = 1, ret.d = 0;
    while(b){
        if(b & 1) ret = mul_er(ret, a, mod);
        a = mul_er(a, a, mod);
        b >>= 1;
    }
    return ret;
}
ll legendre(ll a, ll p){
    return ppow(a, (p - 1) >> 1, p);
}
ll modulo(ll a, ll mod){
    a %= mod;
    if(a < 0) a += mod;
    return a;
}
ll solve(ll n, ll p){   //x^2 = n mod p
    if(n == 0) return 0;
    if(n == 1) return 1;
    if(p == 2) return 1;
    if(legendre(n, p) + 1 == p) return -1;  //无解
    ll a = -1, t;
    while(true){
        a = rand() % p;
        t = a * a - n;
        w = modulo(t, p);
        if(legendre(w, p) + 1 == p) break;
    }
    TT temp;
    temp.p = a;
    temp.d = 1;
    TT ans = power(temp, (p + 1) >> 1, p);
    return ans.p;
}

代码：

#include

转载于:https://www.cnblogs.com/KirinSB/p/11364700.html