其他分享
首页 > 其他分享> > BSGS、exBSGS

BSGS、exBSGS

作者:互联网

BSGS

  \(\text{BSGS}\)(\(\text{Baby Step Giant Step}\))是用来解决\(a^x \equiv b \pmod p\)的问题。

(a,p)=1

  令\(m=\lceil \sqrt{p} \rceil\),则\(x\)可以表示成\(mq+r\),其中\(0 \leq q \leq m\),\(0 \leq r < m\)。

  那么\(a^x\equiv b \pmod p\)可以写成\(a^{mq+r} \equiv b \pmod p\)。

  移项得:\(a^{mq} \equiv ba^{-r} \pmod p\)。

  现在问题变成了:找出一组\(q\),\(r\),使得上式成立,我们枚举\(q\),将左边得到的式子的所有种取值哈希起来,然后再枚举\(r\),判断右边的取值是否在左边的取值中出现,如果出现就找到了解。

  枚举部分的复杂度\(\text{O}(\sqrt p)\),求逆元的复杂度为\(\text{O}(\log p)\),所以总复杂度为\(\text{O}(\sqrt p + \log p)\)。

  如果你不愿意求逆也可以,只要设\(x=mq-r\),然后\(0 < q \leq m\),\(0 \leq r < m\),然后移项过程中\(-r\)变成\(+r\),于是就不用逆元了。

unordered_map<int, int> Hash;
int BSGS(int a, int b) { // a^x=b(P)
	if (!a) return b ? -1 : 1; // 0^x=0的特判
	Hash.clear(); // 清空
	int m = (int)ceil(sqrt(P+0.5)), am = qpow(a, m), t = 1; // 求出m,am
	for (int i = 1; i <= m; i++, t = 1ll*t*am%P) if (!Hash.count(t)) Hash[t] = i; // 将a^mi插入(不可重复插入,否则大的会覆盖小的)
	int ans = -1; t = b;
	rep0(i, m) {
		unordered_map<int, int>::iterator it = Hash.find(t); // 查看是否存在
		if (it != Hash.end()) {
			int x = it->second * m - i;
			if (ans == -1 || x < ans) ans = x;
		}
		t = 1ll*t*a%P;
	}
	return ans; // -1表示无解
}

(a,p)≠1

  这种情况下,\(a\)的逆元不存在,所以我们要约分两边使得\(a\)的逆元出现。

  对于\(a^x \equiv b \pmod p\),提出一个\(d=(a,p)\)并约分得到\(a^{x-1}\frac{a}{d} \equiv \frac{b}{d} \pmod {\frac{p}{d}}\)。当然要有\(d \mid b\),否则无解。

  继续,如果\(d'=(a,\frac{p}{d}) \neq 1\),再拿出一个\(a\)提取因子,得到\(a^{x-2}\frac{a}{d}\frac{a}{d'} \equiv \frac{b}{dd'} \pmod {\frac{p}{dd'}}\)

  以此类推,直到其互质,此时有\(a^{x-t}\frac{a^t}{\prod d} \equiv \frac{b}{\prod d} \pmod {\frac{p}{\prod d}}\)

  显然此时\(\frac{a^t}{\prod d}\)存在逆元,于是可以移过去得到\(a^{x-t} \equiv \frac{b}{\prod d}\frac{a^{-t}}{\prod d} \pmod {\frac{p}{\prod d}}\),然后就成了上面的问题。

  这里约分时由于约分和求公约数,多了一个\(\text{O}(\log^2 p)\)的复杂度,其他不变。

int exBSGS(int a, int b, int P) {
	if (!a) return b ? -1 : 1; // 直接特判
	int t = 0, d, l = 1;
	while (d = gcd(a, P), d > 1) {
		if (b % d) return -1; // 不能整除一定无解
		P /= d, b /= d, l = 1ll*l*(a/d)%P, t++; // 约分,记录指数,记录系数
	}
	a %= P, b = 1ll*b*inv(l, P)%P; // 维护新的a,b
	int ans = BSGS(a, b, P); // 求解
	return ~ans ? ans+t : -1; // 补上t或者无解
}

标签:prod,frac,int,exBSGS,pmod,ans,BSGS,equiv
来源: https://www.cnblogs.com/ac-evil/p/12845250.html