牛客多校第九场 B Quadratic equation 模平方根

题意：

已知

$x+y$ $mod$ $q = b$

$x*y$ $mod$ $q = c$

已知b和c，求x和y

题解：

容易想到$b^2-4c=x^2-2xy+y^2=(x-y)^2$

那么开个根号就能得到x-y，很容易就得出x和y了

在模q意义下对k开根号的方法就是找到w，使得$w*w$ $mod$ $q=k$

考虑模数q为奇质数的情况，可以用Tonelli-Shanks算法求解，这是一个概率算法，但是一般而言得出正确解的概率非常高，遇到类似问题套版即可。

#include<iostream>
#include<cmath>
#define MOD 1000000007
#define LL long long
using namespace std;
LL  qpow(LL x,LL y,LL m){
    //cal x^y%m
        LL re = ;
        while(y){
            if(y & )//判断n的最后一位是否为1
                re = (re * x) % m;
            y >>= ;//舍去n的最后一位
            x = (x * x) % m;//将a平方
        }
        return re % m;
    }
class ModSqrt{
    public:
    LL  power(LL x,LL y,LL m){
    //cal x^y%m
        LL re = ;
        while(y){
            if(y & )//判断n的最后一位是否为1
                re = (re * x) % m;
            y >>= ;//舍去n的最后一位
            x = (x * x) % m;//将a平方
        }
        return re % m;
    }

    LL normal_power(LL a,LL b){
        LL re = ;
        while(b){
            if(b & )//判断n的最后一位是否为1
                re = (re * a);
            b>>= ;//舍去n的最后一位
            a = (a * a);//将a平方
        }
        return re;
    }

    LL inverse(LL a,LL m){
        return power(a,m-,m);
    }

    LL getn(LL p){
        for(LL i=;i<p;i++){
            if(power(i,(p-)/,p)==p-)
                return i;
        }
        return -;
    }

    int solve(LL a,LL p){

        if(a==)return ;
        if(power(a,(p-)/,p)==p-){
            return -;
        }
        while(a<)a+=p;
        while(a>p)a-=p;
        LL n=getn(p);
        bool finish=false;
        LL t=;
        LL s=p-;
        while(!finish){
            s=s/;
            if(s%) finish=true;
            else t+=;
        }
        LL b=power(n,s,p);
        LL _a=inverse(a,p);
        LL x[];
        for(LL i=;i<;i++)x[i]=;
        x[t-]=power(a,(s+)/,p);
        for(LL i=;i<=t-;i++){
            LL judge=power((_a*x[t-i]*x[t-i])%p,normal_power(,t-i-),p);
            if(judge==){
                x[t-i-]=x[t-i];
            }else if(judge==p-){
                x[t-i-]=(power(b,normal_power(,i-),p)*x[t-i])%p;
            }
        }
        return x[];

    }
}modsqrt;
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        LL b,c;
        scanf("%lld %lld",&b,&c);
        LL q=b*b%MOD-*c%MOD;
        q=(q+MOD)%MOD;
        LL sq=modsqrt.solve(q,MOD);
        if(sq==-)printf("-1 -1\n");
        else{
            LL xx=sq+b;
            LL yy=(b-sq+MOD)%MOD;
            LL x=xx*qpow(,MOD-,MOD)%MOD;
            LL y=yy*qpow(,MOD-,MOD)%MOD;
            if(x>y)swap(x,y);
            printf("%lld %lld\n",x,y);
        }
    }
    return ;
}