Timus 1132 Square Root(二次剩余 解法2）

不理解，背板子

#include<cstdio>

using namespace std;

int Pow(int a,int b,int p)
{
    int res=;
    for(;b;a=1LL*a*a%p,b>>=)
        if(b&) res=1LL*a*res%p;
    return res;
}

bool Legendre(int a,int p)
{
    return Pow(a,p->>,p)==;
}

void modsqr(int a,int p)
{
    int x;
    int i,k,b;
    if(p==) x=a%p;
    else if(p%==) x=Pow(a,p+>>,p);
    else
    {
        for(b=;Legendre(b,p);++b);
        i=p->>;
        k=;
        do
        {
            i>>=;
            k>>=;
            if(!((1LL*Pow(a,i,p)*Pow(b,k,p)+)%p)) k+=p->>;
        }while(!(i&));
        x=1LL*Pow(a,i+>>,p)*Pow(b,k>>,p)%p;
    }
    if(p-x<x) x=p-x;
    if(x==p-x) printf("%d\n",x);
    else printf("%d %d\n",x,p-x);
}        

int main()
{
    freopen("data.txt","r",stdin);
    freopen("aa.txt","w",stdout);
    int T;
    scanf("%d",&T);
    int a,n;
    while(T--)
    {
        scanf("%d%d",&a,&n);
        a%=n;
        if(!Legendre(a,n))
        {
            puts("No root");
            continue;
        }
        modsqr(a,n);
    }
    return ;
}