BSGS算法,即Baby Step Giant Step算法,也称作大步小步算法。
给定一个质数 p,以及一个整数 b,一个整数 n,现在要求你计算一个最小的 x,满足 y ^x ≡ n(modp)。
思路 :设 x = am-b, 那么原式可化为 : y ^(am)≡ n * y ^b(modp) 。
那么 0 <= b <= m-1 , 0 <= a <= m+1 。复杂度为 max(m,p/m), 所以当 m = sqrt( p ) 时复杂度最低。
所以我们令 m = sqrt( p ) 。
#include <iostream>
#include <algorithm>
#include <map>
#include <cmath>
using namespace std;
map<int,int> mp;
int main(){
long long p,b,n;
cin >> p >> b >> n;
long long m = sqrt(p);
long long x = n%p;
b %= p; //防止溢出
long long y = 1;
for(long long i = 0;i<m;i++){
mp[x] = i;
x = (x*b)%p;
y = (y*b)%p;
}
long long ans = 1;
for(long long i = 0;i <= m+1;i++){
if(mp.find(ans) != mp.end() ){
cout << i*m-mp[ans] << endl;
return 0;
}
ans = (ans*y)%p;
}
cout << "no solution" << endl;
return 0;
}
代码中的 p, b ,n分别对应公式中的 p, y , n 。 而代码中的 y 即为 公式中的 y ^m 。
此处无需使用快速幂算法。同步迭代即可。