ACM - 数论 | 二次同余方程（质数模）

二次同余方程

形如 $X^2\equiv n$ $(mod$ $p)$ （ $p$ 为质数）形式的方程，我们称之为质数模的二次同余方程 。下面将详细阐述二次同余方程是否有解的判定，并且给出解的形式，和代码求解。

定理

费马小定理

对于一个质数 $p$ 和一个数 $a$ （ $a$ 不为 $p$ 的倍数）有 $a^p$${}^{-}$${}^1\equiv 1$ $(mod$ $p)$ .

证明与结论

结论一

$n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$ $\equiv$ $\pm 1$ $(mod$ $p)$
当值为 $-1$ 时，实际 $n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2=p-1$ .

证明：
由于 $n^p$${}^{-}$${}^1\equiv 1$ $(mod$ $p)$ .

故 $n^p$${}^{-}$${}^1-1\equiv 0$ $(mod$ $p)$ .

故 $($$n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$$+1)($$n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$$-1)$ $\equiv 0$ $(mod$ $p)$ .

故 $n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$ $\equiv$ $\pm 1$ $(mod$ $p)$ .

结论二（欧拉判别条件）

（1）当 $n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2\equiv$ $+1$ $(mod$ $p)$ 时，二次同余方程有解 ；

（2）当 $n^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2\equiv$ $-1$ $(mod$ $p)$ 时，二次同余方程无解 。

结论三

若 $a$ 满足 $w=a^2-n$ 使 $X^2\equiv w(mod$ $p)$ 无解，那么 $(a+sqrt(w))$${}^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$ 即为二次同余方程的解 .

证明：
由于 $a^p$${}^{-}$${}^1\equiv 1$ (mod $p)$ 故 $a^p\equiv a$ $(mod$ $p)$ .

再由于 $p$ 为质数，故 $C_p^i\equiv 0$ $(mod$ $p)$. 同时由条件可得 $w$${}^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2\equiv -1$ $(mod$ $p)$ .

故 式子 $(a+sqrt(w){)}^p$ 经过展开和化简可以得到等式 $(a+sqrt(w){)}^p\equiv a-sqrt(w)$ $(mod$ $p)$ .

将式子两边同乘以$(a+sqrt(w))$ 可得 $(a+sqrt(w){)}^p$${}^{+}$${}^1\equiv a^2-w\equiv n$ $(mod$ $p)$ .

故 $X^2\equiv$ $(a+sqrt(w){)}^p$${}^{+}$${}^1$ 即 $X\equiv$ $(a+sqrt(w))$${}^{(}$${}^p$${}^{-}$${}^1$${}^{)}$${}^{/}$${}^2$ .

代码

说明：
（1） 在代码中，$a$ 利用随机函数获取，由于获得可行的 $a$ 的几率较高，所以在获取 $a$ 的过程中不会耗时过长。

（2） 在处理 $(a+sqrt(w){)}^n$ 的过程中运用到了二次域，其实便是记录 $sqrt(w)$ 的系数来进行幂运算，由于上述的推导，可知最后 $sqrt(w)$ 的系数一定会变为 $0$ , 二次域快速幂只需将普通的快速幂稍作改变即可实现 。

备注： 有些时候 $X$ 可能是一个式子，最终求解未知数 $x$ 的时候可能会遇上除法，这需要看成分数，利用逆元来解决，而不是直接用除法来解。

ll n, w, ans;
int mod = 1e9 + 7;

struct Qd { //二次域
	ll p, d;
	Qd mul(Qd a, Qd b) {
		Qd ans;
		ans.p = ((a.p*b.p) % mod + (a.d*b.d) % mod*w % mod) % mod;
		ans.d = ((a.p*b.d) % mod + (a.d*b.p) % mod) % mod;
		return ans;
	}
	Qd qmul(Qd n, ll m) {
		Qd ans;
		ans.p = 1; ans.d = 0;
		while (m) {
			if (m & 1) {
				ans = ans.mul(ans, n);
			}
			m >>= 1;
			n = n.mul(n, n);
		}
		return ans;
	}
};

ll qpow(ll n, ll m, int mod) { //快速幂
	ll res = 1;
	while (m) {
		if (m & 1) {
			res = (res*n) % mod;
		}
		m >>= 1;
		n = (n*n) % mod;
	}
	return res;
}

ll getans() { //求解 X
	ll a, t;
	do {
		a = rand() % mod;
		t = a*a - n;
		w = (t + mod) % mod;
	} while (qpow(w, (mod - 1) / 2, mod) + 1 != mod);
	Qd tmp;
	tmp.p = a; tmp.d = 1;
	Qd ans = tmp.qmul(tmp, (mod + 1) / 2);
	return ans.p;
}

int main() {
	scanf("%lld", &n);
	if (qpow(n, (mod - 1) / 2, mod) + 1 == mod) {
		cout << "No solution";
	}
	else {
		ans = getans();
		printf("%lld",ans);
	}
	return 0;
}