下文只在模质数意义下讨论。
即给定n>0,p,pn>0,p,pn>0,p,p是质数,求所有x∈[0,p),x2=n(modp)x\in[0,p),x^2=n\pmod px∈[0,p),x2=n(modp)。
若存在解则nnn称为ppp的二次剩余。
首先将p=2p=2p=2的特殊情况判掉。下文ppp是奇质数。
首先显然模奇质数ppp意义下会有恰好p−12\frac{p-1}22p−1个二次剩余。
首先如何判定,有一个定理:
(−1)[n不是p的二次剩余]=np−12(modp)(-1)^{[n不是p的二次剩余]}=n^{\frac{p-1}2}\pmod p(−1)[n不是p的二次剩余]=n2p−1(modp)
并且若有解xxx,则仅有xxx和p−xp-xp−x这两组解。这个定理易证,略去证明。
(上述两个定理都可以通过百度“二次剩余”搜到证明(应该能))
那么考虑一个算法:在[1,p)[1,p)[1,p)随机一个整数ttt,使得:t2−nt^2-nt2−n不是关于ppp的二次剩余。由于有恰好一半不是,因此随机两步就会随机出一组。那么令w=t2−nw=\sqrt{t^2-n}w=t2−n(这里可能不存在在模ppp意义下后半部分的平方根,这里只是定义一个有w2=t2−n(modp)w^2=t^2-n\pmod pw2=t2−n(modp)性质的www)。
可以显然证明:wP=−ww^P=-wwP=−w。
那么考虑令x=(t+w)p+12x=(t+w)^{\frac{p+1}2}x=(t+w)2p+1
x2=(t+w)p+1=(t+w)(t+w)p=(t+w)∑i=0p(pi)tiwp−i=(t+w)(tp+wp)=(t+w)(t−w)=t2−w2=t2−t2+n=nx^2=(t+w)^{p+1}\\=(t+w)(t+w)^p\\=(t+w)\sum_{i=0}^p\binom pit^iw^{p-i}\\=(t+w)(t^p+w^p)\\=(t+w)(t-w)=t^2-w^2\\=t^2-t^2+n=nx2=(t+w)p+1=(t+w)(t+w)p=(t+w)∑i=0p(ip)tiwp−i=(t+w)(tp+wp)=(t+w)(t−w)=t2−w2=t2−t2+n=n
因此xxx就是x2=n(modp)x^2=n\pmod px2=n(modp)的一组解,其解集为{x,p−x}\{x,p-x\}{x,p−x}
时间复杂度O(lgp)O(\lg p)O(lgp)。实现的时候可以将(t+w)k(t+w)^k(t+w)k视为v+cwv+cwv+cw,利用w2=t2−nw^2=t^2-nw2=t2−n即可做一次乘法。代码(应该是正确的):
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define gc getchar()
using namespace std;typedef long long lint;
inline int inn() { int x,ch;while((ch=gc)<'0'||ch>'9');x=ch^'0';while((ch=gc)>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^'0');return x; }
inline int fast_pow(int x,int k,int p,int ans=1) { for(;k;k>>=1,x=(lint)x*x%p) (k&1)?ans=(lint)ans*x%p:0;return ans; }
struct RAND{ unsigned int x;RAND() { x=1000000007; }
inline int operator() (int a,int b) { return x=(x<<11)^(x>>3)^x^998244353,int(x%(b-a+1)+a); }
}rnd; struct E{ int t,c;E(int _t=0,int _c=0) { t=_t,c=_c; } };
inline E tms(const E &a,const E &b,int w2,int p) { return E(((lint)a.t*b.t+(lint)a.c*b.c%p*w2)%p,((lint)a.c*b.t+(lint)a.t*b.c)%p); }
inline int calc(E x,int w2,int k,int p) { E ans(1,0);for(;k;k>>=1,x=tms(x,x,w2,p)) if(k&1) ans=tms(ans,x,w2,p);return ans.t; }
inline int solve(int n,int p)
{
if(fast_pow(n,(p-1)/2,p)==p-1) return -1;int t,w;
do{
t=rnd(1,p-1),w=((lint)t*t-n)%p;if(w<0) w+=p;
}while(fast_pow(w,(p-1)/2,p)==1);
return calc(E(t,1),w,(p+1)/2,p);
}
int main()
{
for(int T=inn();T;T--)
{
int n=inn(),p=inn();n%=p;if(!n) { printf("0\n");continue; }
if(p==2) { printf("1\n");continue; }int ans1=solve(n,p),ans2=p-ans1;
if(ans1==-1) { printf("No answer!\n");continue; }
if(ans1>ans2) swap(ans1,ans2);printf("%d %d\n",ans1,ans2);
}
return 0;
}
`