【模板】BSGS

最新推荐文章于 2024-07-13 23:05:26 发布

BMTXLRC

最新推荐文章于 2024-07-13 23:05:26 发布

阅读量78

点赞数

分类专栏：数论-普通文章标签：算法

本文链接：https://blog.csdn.net/BMTXLRC/article/details/127228823

版权

数论-普通专栏收录该内容

12 篇文章 0 订阅

订阅专栏

【模板】BSGS

我们一般叫他北上广深算法。

Part 1. BSGS 是干什么的

求解如下问题：

已知 $a, b, p$ ，求最小的正整数 $x$ 使得 $a^x\equiv b\pmod p$ ，保证 $p$ 是质数。

Part 2. 做法

由欧拉定理我们可以知道：
$a^{\varphi(p)}\equiv 1\pmod p$

又因为：
$a^0\equiv 1\pmod p$

所以可以发现 $[0,\varphi(p)-1]$ 为 $a_i$ 的循环节，考虑枚举一定范围来求解。

我们设 $x=ik-j(0\leqslant j<k)$ ，可以得到 $a^{ik-j}\equiv b\pmod p$ ，转化为：
$a^{ik}\equiv ba^j\pmod p$

于是我们考虑将所有的 $ba^j\bmod p$ 求出来，显然有 $k$ 个，塞入哈希表里，枚举 $i\in[0,k]$ ，查找对应的 $ba^j$ 值，即可得到答案。
现在我们的复杂度是 $\mathcal{O}\left(\max\left(k,\dfrac{\varphi(p)}{k}\right)\right)$ ，由基本不等式取 $k=\lceil\sqrt{p}\rceil$ 时时间复杂度最优。

特别要注意的是，一定不能取在 $\lfloor\sqrt{p}\rfloor$ ，要么向上取整，要么使用向下取整 $+ 1$ 。
如果我们使用 map 进行哈希表处理可以得到时间复杂度 $\mathcal{O}(\sqrt{p}\log p)$ ，用普通的哈希表或者 unordered_map 可以做到 $\mathcal{O}(\sqrt p)$ 。

#include<bits/stdc++.h>
#define int long long
using namespace std;
inline int read(){
    char op=getchar();
    int w=0,s=1;
    while(op<'0'||op>'9'){
        if(op=='-') s=-1;
        op=getchar();
    }
    while(op>='0'&&op<='9'){
        w=(w<<1)+(w<<3)+op-'0';
        op=getchar();
    }
    return w*s;
}
int mod;
int Mul(int a,int b){return (a%mod*b%mod)%mod;}
int Add(int a,int b){return (a+b)%mod;}
int Dec(int a,int b){return (a-b+mod)%mod;}
int Pow(int a,int k){
    int ans=1;
    while(k){
        if(k&1) ans=Mul(ans,a);
        a=Mul(a,a);
        k>>=1;
    }
    return ans;
}
map<int,int> mp;
signed main(){
    int p=read(),a=read(),b=read();
    mod=p;
    int k=sqrt(p)+1;
    for(register int i=0,cnt=b;i<=k;i++,cnt=Mul(cnt,a)) mp[cnt]=i;
    int now=Pow(a,k);
    for(register int i=1,cnt=now;i<=k;i++,cnt=Mul(cnt,now)){
        if(mp.count(cnt)!=0){
            printf("%lld",i*k-mp[cnt]);
            return 0;
        }
    }
    printf("no solution");
}