洛谷-5491 【模板】二次剩余

题目描述
求解方程 $x^2\equiv N(\mathrm{m}\mathrm{o}\mathrm{d}p)$
多组数据。
保证p是奇素数
输入格式
第一行一个整数T表示数据组数。
接下来T行，每行一个N一个p。
输出格式
输出共T行。
对于每一行输出：
若有解，则按$\mathrm{m}\mathrm{o}\mathrm{d}p$ 后递增的顺序输出在$\mathrm{m}\mathrm{o}\mathrm{d}p$意义下的全部解.
若两解相同，只输出其中一个；
若无解，则输出Hola!；

输入输出样例
输入 #1
3
5 1000000009
4 1000000009
0 19260817

输出 #1
383008016 616991993
2 1000000007
0

说明/提示
$T\le 10000$
$N,p\le 10^9+9$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll w;
struct num{
    ll x,y;
};
num mul(num a,num b,ll p){
    num ans={0,0};
    ans.x=((a.x*b.x%p+a.y*b.y%p*w%p)%p+p)%p;
    ans.y=((a.x*b.y%p+a.y*b.x%p)%p+p)%p;
    return ans;
}
ll powwR(ll a,ll b,ll p){
    ll ans=1;
    while(b){
        if(b&1)ans=1ll*ans%p*a%p;
        a=a%p*a%p;
        b>>=1;
    }
    return ans%p;
}
ll powwi(num a,ll b,ll p){
    num ans={1,0};
    while(b){
        if(b&1)ans=mul(ans,a,p);
        a=mul(a,a,p);
        b>>=1;
    }
    return ans.x%p;
}
ll solve(ll n,ll p){
    n%=p;
    if(p==2)return n;
    if(powwR(n,(p-1)/2,p)==p-1)return -1;//不存在
    ll a;
    while(1){
        a=rand()%p;
        w=((a*a%p-n)%p+p)%p;
        if(powwR(w,(p-1)/2,p)==p-1)break;
    }
    num x={a,1};
    return powwi(x,(p+1)/2,p);
}
int main(){
    srand(time(0));
    int t;
    scanf("%d",&t);
    while(t--){
        ll n,p;
        scanf("%lld%lld",&n,&p);
        if(!n){
            printf("0\n");continue;
        }
        ll ans1=solve(n,p),ans2;
        if(ans1==-1)printf("Hola!\n");
        else{
            ans2=p-ans1;
            if(ans1>ans2)swap(ans1,ans2);
            if(ans1==ans2)printf("%lld\n",ans1);
            else printf("%lld %lld\n",ans1,ans2);
        }
    }
}