【算法讲25：BSGS】BSGS与扩展BSGS

本文链接：https://blog.csdn.net/weixin_45775438/article/details/118683753

本文介绍了BSGS（Baby Step Giant Step）算法及其扩展版ExBSGS，用于求解模指数方程。在原始BSGS算法中，当gcd(a,p)=1时，通过平方根优化暴力枚举。而在ExBSGS中，当gcd(a,p)≠1时，通过提取最大公约数并递归处理，解决了非逆元情况下的模指数问题。博客提供了递归和非递归两种ExBSGS算法的C++实现，并给出了样例测试。

摘要由CSDN通过智能技术生成

【算法讲25：BSGS】

引入
BSGS
代码
扩展BSGS
ExBSGS
递归版代码
非递归版

yjw的博客入口

引入

P3846 [TJOI2007] 可爱的质数/【模板】BSGS
给定一个质数 $p$ 以及整数 $a 、 b$
你需要给出最小的非负整数 $x$ ，满足：
$a^x\equiv b\pmod p$
或输出无解。
$2\le a,b<p<2^{31}$

BSGS

全称为 $Baby\ Step\ Giant\ Step$ ，简称BSGS或者北上广深算法。
怎么做呢？其实本质是枚举
最原始的做法：暴力枚举 $x\in[0,p)$ 然后检验答案是否正确。
优化，我们可以考虑折半枚举。
我们令 $x = k n + i$ ，这里 $n$ 是我们设的一个定值， $k 、 i$ 都为变量。
这样，原式变成： $a^{kn+i}\equiv b\pmod p$
由于 $\gcd(a,p)=1$ ，我们转移一下变成： $a^{kn}\equiv b(a^i)^{-1}\pmod p$
我们可以枚举所有的 $i$ ，将右边的值存入哈希表。
然后枚举 $k$ 的过程中，去表中找满足的 $i$ ，找到后 $k n + i$ 就是答案。
那么 $n$ 的设置，就像分块思想一样，我们设为 $\sqrt p$ ，这样 $k 、 i$ 都只用枚举到 $\sqrt p$

代码

时间复杂度： $O(\sqrt p)$

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 30;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll qpow(ll a,ll n,ll p){a%=p;ll res = 1LL;while(n){if(n&1)res=res*a%p;a=a*a%p;n>>=1;}return res;}
ll exgcd(ll a,ll b,ll& x,ll& y){
    if(b==0){
        x=1;y=0;
        return a;
    }
    ll ans = exgcd(b,a%b,x,y);
    ll tmp = x;
    x = y;
    y = tmp - a/b*y;
    return ans;
}
ll inv(ll a,ll mod){
    ll x,y;
    ll g = exgcd(a,mod,x,y);
    if(g!=1)return -1;
    return (x%mod+mod)%mod;
}
ll BSGS(ll a,ll b,ll p){
    b%=p;
    if(b==1||p==1)return 0;
    ll n = sqrt(p);
    unordered_map<ll,ll>M;
    ll inva = inv(qpow(a,n-1,p),p) * b % p;
    for(ll i = n-1;~i;--i){
        M[inva] = i;
        inva = inva * a % p;
    }
    ll ta = 1,powa = qpow(a,n,p);
    for(ll k = 0;k <= p;k+=n){
        if(M.count(ta))return k + M[ta];
        ta = ta * powa % p;
    }
    return -1;
}
int main()
{
    ll p,b,n,ans;
    cin >> p >> b >> n;
    ans = BSGS(b,n,p);
    if(ans == -1)puts("no solution");
    else cout << ans;
    return 0;
}

扩展BSGS

P4195 【模板】扩展 BSGS/exBSGS
给定整数 $a 、 b 、 p$
你需要给出最小的非负整数 $x$ ，满足：
$a^x\equiv b\pmod p$
或输出无解。
$1\le a,b,p<10^9$

ExBSGS

因为 $\gcd(a,p)\ne 1$ ，所以不能像原来一样使用逆元了。
所以我们想把 $g c d$ 先提出来。我们令 $d=\gcd(a,p)$
我们把原式 $a^x\equiv b\pmod p$ 展开成 $a^x+kp=b$
两边同时除以 $d$ ，变成 $\frac{a^x}{d}+k\frac{p}{d}=\frac{b}{d}$
取模，得到： $\frac{a^x}{d}\equiv \frac{b}{d}\pmod {\frac{p}{d}}$
发现， $b$ 必须是 $d$ 的倍数，否则直接返回无解。
此时 $\gcd(\frac{a}{d},\frac{p}{d})=1$ ，所以就可以乘上它的逆元
变成： $a^{x-1}\equiv \frac{b}{d}(\frac{a}{d})^{-1}\pmod {\frac{p}{d}}$
然后就变成了一个新的问题。但是此时 $\gcd(a,\frac{p}{d})$ 不一定为 $1$ ，可以递归处理。（当然也可以不递归处理）
注意就是BSGS里面的 $unordered\_map$ 改成 $m a p$ 在这道题里会快很多，因为出题人特意卡了一个样例…

递归版代码

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 30;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll qpow(ll a,ll n,ll p){a%=p;ll res = 1LL;while(n){if(n&1)res=res*a%p;a=a*a%p;n>>=1;}return res;}
ll exgcd(ll a,ll b,ll& x,ll& y){
    if(b==0){
        x=1;y=0;
        return a;
    }
    ll ans = exgcd(b,a%b,x,y);
    ll tmp = x;
    x = y;
    y = tmp - a/b*y;
    return ans;
}
ll inv(ll a,ll mod){
    ll x,y;
    ll g = exgcd(a,mod,x,y);
    if(g!=1)return -1;
    return (x%mod+mod)%mod;
}
ll BSGS(ll a,ll b,ll p){
    b%=p;
    if(b==1||p==1)return 0;
    ll n = sqrt(p);
    unordered_map<ll,ll>M;
    ll inva = inv(qpow(a,n-1,p),p) * b % p;
    for(ll i = n-1;~i;--i){
        M[inva] = i;
        inva = inva * a % p;
    }
    ll ta = 1,powa = qpow(a,n,p);
    for(ll k = 0;k <= p;k+=n){
        if(M.count(ta))return k + M[ta];
        ta = ta * powa % p;
    }
    return -1;
}
ll ExBSGS(ll a,ll b,ll p){
    b%=p;
    if(a==0 && b==0)return 1;
    if(a==0 && b!=0)return -1;
    if(b==1||p==1)return 0;
    ll d = __gcd(a,p);
    if(b%d!=0)return -1;
    p/=d;
    b=b/d*inv(a/d,p)%p;
    if(d!=1){
        ll ans = ExBSGS(a,b,p);
        if(ans==-1)return -1;
        return ans+1;
    }
    ll ans = BSGS(a,b,p);
    if(ans == -1)return -1;
    return ans+1;
}
int main()
{
    ll a,b,p,ans;
    while(~scanf("%lld%lld%lld",&a,&p,&b)){
        if(p == 0)break;
        ans = ExBSGS(a,b,p);
        if(ans == -1)puts("No Solution");
        else printf("%lld\n",ans);
    }

    return 0;
}

非递归版

ll ExBSGS(ll a,ll b,ll p){
    b%=p;
    if (b==1||p==1)return 0;
    int g=__gcd(a,p),k=0,na=1;
    while (g>1)
    {
        if (b%g!=0)return -1;
        k++;b/=g;p/=g;na=na*(a/g)%p;
    	if (na==b)return k;
    	g=__gcd(a,p);
    }
    int f=BSGS(a,b*inv(na,p)%p,p);
    if (f==-1)return -1;
    return f+k;
}