数论之高次同余方程(Baby Step Giant Step + 拓展BSGS)
什么叫高次同余方程?说白了就是解决这样一个问题:
A^x=B(mod C),求最小的x值。
baby step giant step算法
题目条件:C是素数(事实上,A与C互质就可以。为什么?在BSGS算法中是要求a^m在%c条件下的逆元的,如果a、c不互质根本就没有逆元。)
如果x有解,那么0<=x<C,为什么?
我们可以回忆一下欧拉定理:
对于c是素数的情况,φ(c)=c-1
那么既然我们知道a^0=1,a^φ(c)=1(在%c的条件下)。那么0~φ(c)必定是一个循环节(不一定是最小的)。既然是%c,那么B一定是0到c-1之间的一个数。最坏的条件下,a的φ(c)以内次方%c的余数各不相同,0<=x<C时一定存在一个x满足条件。根据抽屉原理,如果在大于c的x中出现一个x值满足条件,那在它以前一定有一个更小的x值满足条件。
BSGS的算法是这样的:
首先取m=sqrt(c)向上取整。(为什么取sqrt(c)?我也不是很懂,是为了算法的效率平衡。)
然后先预处理a的0到m次方。
a^x=b ( %c )
设x=i*m+j;
即: i为x/m,j为x%m。
a^(i*m+j)=b;
b * (a^(-m))^i = a^j ( %c )
先枚举j,把右边存起来(Hash 或者 普通数组,下一步用二分查找)
枚举i,如果左边的数值曾经存储过(b * (a^(-m))^i = a^j),则 x=i*m+j。
求a^(-m):(就是a^m的逆元)
有两种方法:
方法一:根据欧拉定理
设A=a^m,那么A^φ(c)==1(%c)
A^(φ(c)-1)*A==1(%c)
到这里已经可以得到A的逆元为A^(φ(c)-1)。
继续推下去,根据c是素数,φ(c)=c-1
那么A的逆元就是A^(c-2)
方法二:相当于解a^m*x-C*y=1,根据拓展欧几里得出x就是逆元。
BSGS主要就是要注意细节,注意要去重(余数相同时只要取较小的一个)。
拓展BSGS
如果a跟c不互质,那该怎么办?
其实只需要加一小段代码就可以。
首先,我们知道:
A%C=B,那么就是A-C*x=B,如果d=gcd(A,C),且B%d==0,那么(A/d)%(C/d)=B/d是成立的。
那么我们就在A与C仍有不为一的公因数的时候,不断地从a^x中拿出一个a与c约分。过程中如果b%d!=0,那么在x>T的时候无解。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
LL D=1%C; LL g=0,d;
while( ( d=gcd(A,C) ) !=1 )
{
if(B%d)return -1;
B/=d;C/=d;
g++;D=D*(A/d)%C;
}
最后我们的方程就变为了k*a^(x-g) == b' (%c')
用BSGS解出x后加上g就是答案。