二次剩余

给定两个数$n,P$,$P$为一个质数，且$n < P$，求一个$x$，使得$x^2 \equiv n (\mathrm{mod}~P)$

下面所有的运算都在$\mathrm{mod}~P$意义下进行。

首先我们要判断是否有解。

根据勒让德记号，我们有$n^{\frac{P - 1}{2}} = \pm1$ ，并且上述问题有解当且仅当$n^{\frac{P - 1}{2}} = 1$

设$a$，满足$w = a^2-n$，$w$对于$P$不是二次剩余。这个$a$直接随机就好了。

那么最终我们有$x = (a + \sqrt{w})^{\frac{P + 1}{2}}$

问题是$\sqrt{w}$不存在，我们将$\sqrt{w}$如同虚数$i$一样设定，那么由于最终$x$是一个实数，所以直接当虚数来做快速幂最终取实根即可。

证明：
$x = (a + \sqrt{w})^{\frac{P+1}{2}}$
$x^2 = (a + \sqrt{w})^{P+1} = (a + \sqrt{w})^P * (a + \sqrt{w})$

由二项式定理得：
$(a+\sqrt{w})^P = \sum_{k=0}^{P} C_{P}^k * a ^ k* (\sqrt{w})^{P - k}$

又由于当$0 < k < P$时，$C_{P}^k~\mathrm{mod}~P = 0$，所以

$$(a + \sqrt{w})^P = a^P + (\sqrt{w})^P$$

又由费马小定理$a^{P - 1} = 1,w^{\frac{P - 1}{2}} = \sqrt{w}^{P - 1} = -1$

所以

$$(a + \sqrt{w})^P = a - \sqrt{w}$$

所以$x^2 = (a - \sqrt{w})(a + \sqrt{w}) = a^2 - w = n$

得证