BSGS与exBSGS学习笔记

\(BSGS\)用于解决这样一类问题:

求解\(A^x ≡B(modP)\)的最小\(x\),其中\(P\)为质数。

这里我们采用分块的方法,把\(x\)分解为\(i *t-b\)(其中\(t\)是分块大小) 。根据模意义下逆元的性质,\(x\)的大小一定\(<=phi(p)\)即\(p - 1\),所以经过移项和进行存在性对比,我们就可以\(O(N)\)求出答案。

int BSGS (int A, int B, int P) {
int t = (int) ceil (sqrt (P));
for (int j = 0; j < t; ++j) {
mp[_mul (B, _pow (A, j, P), P)] = j;
//mp[ B * A ^ j ] = j;
}
A = _pow (A, t, P);
for (int i = 1; i <= t; ++i) {
int val = _pow (A, i, P);
//val = A^{i*t};
int j = mp.find (val) == mp.end () ? -1 : mp[val];
if (j >= 0) {
return i * t - j;
}
}
return -1;
}

上面这份代码中其实还可以把快速幂的\(log\)优化掉,可能会被卡常。

几个要注意的关键点:

  • 避免快速幂
  • 优化快速乘
  • 小心取模和\(longlong\)
  • \(p\)一定要是质数!
  • 建议手写哈希不然会多一个\(log\)(\(unordered\_map\)是不允许使用的)

\(exBSGS\)其实就是一个简单的扩展,把情况扩展到了\(p\)不是质数的情况,这种情况我们要先把\(P\)和\(A\)化为互质的状态。也就是说:对\(A\)和\(P\)取\(gcd\)直到其互质为止,从而化为如下形式:

\[(A/d)^{cnt} * A^{x}≡B/d^{cnt} (mod P/d^{cnt})
\]

其中,当\(B\)不能被二者的\(gcd\)整除时,就意味着原方程无解。

几个注意点:

  • 前面的\((A/d)^{cnt}\)同样需要统计进去
  • 要使用\(exgcd\)求解,因为可能不存在逆元。
  • 可能存在\(x==0\)的情况,记得特判
  • 快速乘,\(longlong\),取模,务必小心。
int exbsgs (int A, int B, int p) {
if (B == 1) return 0;
int _gcd, cnt = 0, res = 1;
while ((_gcd = gcd (A, p)) != 1) {
if (B % _gcd) return -1;
B /= _gcd, p /= _gcd, ++cnt;
res = ((res % p) * (A / _gcd)) % p;
if (res == B) return cnt;
}
int t = sqrt (p) + 1, tmp = 1;
Hash.clear ();
for (int i = 0; i < t; ++i) {
Hash[(tmp * B) % p] = i;
tmp = (tmp * A) % p;
}
res = (res * tmp) % p;
for (int i = 1; i <= t; ++i) {
if (Hash.find (res) != Hash.end ()) {
return i * t - Hash[res] + cnt;
}
res = (res * tmp) % p;
}
return -1;
}
上一篇:JavaScript 中 正则替换 replace


下一篇:Maven系列一pom.xml 配置详解