BZOJ 3239 Discrete Logging BSGS

最新推荐文章于 2018-08-08 22:02:53 发布

YihAN_Z

最新推荐文章于 2018-08-08 22:02:53 发布

阅读量345

点赞数

分类专栏： BSGS

本文链接：https://blog.csdn.net/YihAN_Z/article/details/64128603

版权

BSGS 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目大意：对于这里写图片描述，给定a,b,p，求x，其中p为质数。

裸Baby-step-Giant-step

这里写的还不错

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <map>
using namespace std;
typedef long long LL;
LL f_pow(LL x,LL y,LL MOD) {
    LL tmp=1;
    while(y) {
        if(y&1) (tmp*=x)%=MOD;
        (x*=x)%=MOD;
        y>>=1;
    }
    return tmp;
}
map<LL,LL> h;
LL BSGS(LL a,LL b,LL p){
    h.clear();
    LL m=LL(sqrt(p)+1);
    for(int i=0;i<m;i++) {
        if(!h.count(b)) h[b]=i;
        b=b*a%p;
    }
    LL now=1,base=f_pow(a,m,p);
    for(int i=1;i<=m+1;i++) {
        now=now*base%p;
        if(h.count(now)) return i*m-h[now];
    }
    return -1;
}
int main() {
    LL B,P,N;
    while(scanf("%lld%lld%lld",&P,&B,&N)==3) {
        LL ans=BSGS(B,N,P);
        if(ans==-1) printf("no solution\n");
        else printf("%lld\n",ans);
    }
    return 0;
}