离散对数

问题

给出\(a,b,p\)，求最小的\(x\)使得\(a^x\equiv b\mod p\)。

BSGS算法

Baby-Steps-Giant-Steps算法可以在\(\sqrt p\)的时间内求解这个问题，它利用了两个不同步长的遍历。
\[ \begin{aligned} 令m&=\sqrt p \\ a^{im-j}&\equiv b \mod p \\ a^{im}&\equiv ba^j, i\in[1,m],j\in[0,m] \end{aligned} \]
首先求出所有\(ba^j,j\in[0,m]\)，存入哈希表中。若哈希表中已有相同的\(ba^j\)，那么我们更新到\(j\)比较大的那个。再从小到大枚举\(i\)，在哈希表中查询\(a^{im}\)是否存在即可。由于我们要令\(x\)最小，而\(x=im-j\)，所以要令\(j\)最大，\(i\)最小。

这个算法是成立的，原因是，对于朴素的枚举，我们枚举\(x\in [0,p-1]\)即可，而这个算法利用了中途相遇的思想，相当于是枚举了这些值。

离散对数是一个重要的工具，在求解模意义下的方程中很有用。

代码

#include<cstdio>
#include<cctype>
#include<map>
#include<cmath>
#include<cstring>
using namespace std;
typedef long long giant;
int read() {
    int x=0,f=1;
    char c=getchar();
    for (;!isdigit(c);c=getchar()) if (c=='-') f=-1;
    for (;isdigit(c);c=getchar()) x=x*10+c-'0';
    return x*f;
}
const int q=1e9+7;
int mi(int x,int y) {
    int ret=1;
    for (;y;y>>=1,x=(giant)x*x%q) if (y&1) ret=(giant)ret*x%q;
    return ret;
}
const int maxn=1e5+7;
const int maxm=1e7+1;
struct node {
    int val,x,nxt;
};
struct HASH {
    int h[maxn],tot;
    node e[maxm];
    void init() {
        memset(h,0,sizeof h);
        tot=0;
    }
    void insert(int v,int j) {
        int t=v%maxn;
        for (int i=h[t];i;i=e[i].nxt) if (e[i].val==v) {
            e[i].val=v;
            e[i].x=j;
            return;
        }
        e[++tot]=(node){v,j,h[t]};
        h[t]=tot;
    }
    int find(int x) {
        int t=x%maxn;
        for (int i=h[t];i;i=e[i].nxt) if (e[i].val==x) return e[i].x;
        return -1;
    }
} has;
int main() {
    #ifndef ONLINE_JUDGE
        freopen("test.in","r",stdin);
    #endif
    int m=(int)sqrt(q)+1;
    int T=read();
    while (T--) {
        int a=read(),b=read();
        int am=mi(a,m);
        has.init();
        for (int j=0,tmp=b;j<=m;++j,tmp=(giant)tmp*a%q) {
            has.insert(tmp,j);
        } 
        int x=-1;
        for (int i=1,tmp=am;i<=m;++i,tmp=(giant)tmp*am%q) {
            int y=has.find(tmp);
            if (~y) {
                x=i*m-y;
                break;
            }
        }
        printf("%d\n",x);
    }
    return 0;
}

转载于:https://www.cnblogs.com/owenyu/p/6724627.html