二次剩余（模板）

二次剩余

存在 \(x^2\ \%\ p = n\)， 如果存在这样的\(x\)那么，认为\(( \dfrac{n}{p})\) 有解。

• \((\dfrac{n}{p}) = 1\)： n在模p意义下是二次剩余

• \((\dfrac{n}{p}) = -1\)：a在模p意义下是非二次剩余

• \((\dfrac{n}{p}) = 0\) ：\(n\ \% \ p = 0\)

若\(p\)是一个奇素数 ， \((\dfrac{n}{p}) = n^ {\frac{p-1}{2} }\)

步骤

因为有解的情况下， 大约有\(\dfrac{p-1}{2}\)个\(a\)满足条件，大约随机数算出的期望是\(2\)， 所以就可以放心的随机数了

• \(a = rand() \% p\)
• \(w = a*a - n\)
• \((\dfrac{w}{p}) = w^ {\frac{p-1}{2} }= -1\)： \(x = (a +\sqrt{w})^{\frac{p+1}{2}}\)为\((\dfrac{n}{p})\)的解

因为要计算\((a +\sqrt{w})^{\frac{p+1}{2}}\), 可以把\((a +\sqrt{w})^{\frac{p+1}{2}}\)看成两部分来乘（类似于虚数乘法，把\(\sqrt{w}\)看成\(i\))

代码

ll w;
struct Num{ 
    ll x, y;
    Num(){
        x = y = 0;
    }
    Num(ll xx, ll yy){
        x = xx;
        y = yy;
    }
    Num operator * (Num const & a)const{
        Num ans;
        ans.x = ((x * a.x)%p + (y * a.y)%p * w %p + p)%p;
        ans.y = ((x * a.y)%p + (y * a.x)%p + p)%p;
        return ans;
    }
};

ll pow_num(Num a, ll b){    
    Num ans = {1, 0};
    while(b){
        if(b&1)
            ans = ans*a;
        a = a*a;
        b /= 2;
    }
    return ans.x%p;
}

ll pow(ll a, ll b){
    ll ans = 1;
    while(b){
        if(b&1)
            ans = ans*a%p;
        a = a*a%p;
        b /= 2;
    }
    return ans;
}

ll check(ll n){
    ll ans = pow(n, (p-1)/2);
    if(ans == p-1)
        return -1;
    else
        return 1;
}

ll solve(ll n){
    n %= p;
    if(check(n) == -1)  \\如果当前的check(n) == -1那么就不存在解，返回-1
        return -1;
    if(p == 2) return n;    \\p = 2的时候， n = 0，1。那么x = n
    ll a;
    srand(time(NULL));
    while(1){
        a = rand()*13331%p;     \\随机数一个a
        w = ((a*a%p - n)%p + p)%p;  
        if(check(w) == -1)
            break;
    }
    Num ans = {a, 1};
    return pow_num(ans, (p+1)/2);
}