BSGS exBSGS学习笔记

BSGS

BSGS，中文名大步小步算法，用于在 $O(\sqrt{p})$ 时间内求解这样一个同余式：

$a^x\equiv b(\mathrm{mod}p)$，其中$a$，$b$，$p$ 已知且 $p$ 为质数。

算法思想：设 $t=\lceil \sqrt{p}\rceil$，由于若有解，则最小非负整数解 $x$ 一定在 $[0,p]$ 内，所以 $x$ 一定可以被表示为形如 $i\times t-j(i\in [1,t],j\in [0,t])$ 的形式，由此可以将同余式变形：

$a^{i\times t-j}\equiv b(\mathrm{mod}p)\Rightarrow (a^t{)}^i\equiv b\times a^j(\mathrm{mod}p)$

于是我们可以先枚举 $j$，将 $b\times a^j$ 的值插入哈希表或map中与 $j$ 对应，再枚举 $i$ 进行查询，若在哈希表或map中找到了 $(a^t{)}^i$ 的值，则其中一个答案为 $i\times t-j$，若顺序枚举 $i$，则第一个找到的合法解就是最小解。

代码里判了三种无解情况，并用unordered_map优化了一下：

inline int bsgs(int a,int b,int p){
	b%=p;
	if(a%p==0&&b) return -1;
	int t=ceil(sqrt(p)),x=b;
	ff(j,0,t){
		mp[x]=j;
		x=1ll*x*a%p;
	}
	int base=ksm(a,t,p),now=1;
	if(!base) return b==0?1:-1;
	ff(i,1,t){
		now=1ll*now*base%p;
		if(mp[now]) return i*t-mp[now];
	}
	return -1;
}

（洛谷的数据真的水水水啊

exBSGS

当 $a$ 和 $p$ 不互质的时候，想利用BSGS解方程，就要考虑怎么把 $a$ 和 $p$ 变为互质：

设 $g=\gcd (a,p)$，则原式可以转化成 $\frac{a}{g}\times a^{x-1}\equiv \frac{b}{g}(\mathrm{mod}\frac{p}{g})$，那么根据学数学摸爬滚打出的经验，当 $b\mathrm{mod}g\ne 0$且 $b\ne 1$时无解（$b=1$ 的时候只要 $x=0$ 就可以了），这样一直将 $b$ 和 $p$ 除下去，最终 $a$ 和 $p$ 一定互质。

设每次操作得到的的 $\frac{a}{g}$ 累乘起来为 $k$，最终一共操作了 $cnt$ 次，则当前方程变为 $k\times a^{x-cnt}\equiv b(\mathrm{mod}p)$，由于互质，逆元一定存在，可以用扩欧求出 $k$ 的逆元 $inv$，方程又变为 $a^{x-cnt}\equiv b\times inv(\mathrm{mod}p)$，可以BSGS求解。

inline void exbsgs(int a,int b,int p){
	int g=__gcd(a,p),k=1,cnt=0;
	if(p==1||b==1) return printf("0\n"),void();
	while(g!=1){
		if(b==k) return printf("%lld\n",cnt),void();
		if(b%g) return printf("No Solution\n"),void();
		++cnt,b/=g,p/=g,k=k*a/g%p;
		g=__gcd(a,p);
	}
	int inv,y;
	exgcd(k,p,inv,y);
	inv=(inv%p+p)%p;
	b=b*inv%p;
	int res=bsgs(a,b,p);
	if(res==-1) printf("No Solution\n");
	else printf("%lld\n",res+cnt);
}

这个代码加上一个BSGS在洛谷过不去，一直TLE#11，数据加强成这样就很真的绝了/fn

不过理解万岁啦。