【EX_BSGS】BZOJ1467 Pku3243 clever Y

1467: Pku3243 clever Y

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 425  Solved: 238
[Submit][Status][Discuss]

Description

小Y发现，数学中有一个很有趣的式子：
X^Y mod Z = K
给出X、Y、Z，我们都知道如何很快的计算K。但是如果给出X、Z、K，你是否知道如何快速的计算Y呢？

Input

本题由多组数据（不超过20组），每组测试数据包含一行三个整数X、Z、K（0 <= X, Z, K <= 109）。
输入文件一行由三个空格隔开的0结尾。

Output

对于每组数据：如果无解则输出一行No Solution，否则输出一行一个整数Y(0 <= Y < Z)，使得其满足XY mod Z = K，如果有多个解输出最小的一个Y。

Sample Input

5 58 33
2 4 3
0 0 0

Sample Output

9
No Solution

题解

EX_BSGS模板题

思路懒得写了

代码

//by 减维
#include<set>
#include<map>
#include<queue>
#include<ctime>
#include<cmath>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define il inline
#define rg register
#define db double
#define mpr make_pair
#define maxn
#define inf (1<<30)
#define eps 1e-8
#define pi 3.1415926535897932384626L
using namespace std;

inline int read()
{
    int ret=;bool fla=;char ch=getchar();
    while((ch<''||ch>'')&&ch!='-')ch=getchar();
    if(ch=='-'){fla=;ch=getchar();}
    while(ch>=''&&ch<=''){ret=ret*+ch-'';ch=getchar();}
    return fla?-ret:ret;
}

ll x,y,p;
map<ll,ll> mp;

ll gcd(ll x,ll y){return y==?x:gcd(y,x%y);}

ll ksm(ll x,ll y,ll p)
{
    ll ret=;x%=p;
    for(;y;y>>=,x=x*x%p)
        if(y&) ret=ret*x%p;
    return ret;
}

ll exbsgs(ll x,ll y,ll p)
{
    x%=p;y%=p;
    if(!x&&!y) return ;
    if(y==) return ;
    if(!x) return -;
    ll t,d=,k=;
    while((t=gcd(p,x))!=)
    {
        if(y%t) return -;
        y/=t,p/=t,d=d*x/t%p,k++;
        if(d==y) return k;
    }
    mp.clear();
    ll m=sqrt(p+0.5);
    ll o=y;
    for(int i=;i<=m;++i,o=o*x%p)
        if(!mp.count(o)) mp[o]=i;
    ll tmp=ksm(x,m,p);
    for(int i=;i<=m;++i)
    {
        d=d*tmp%p;
        if(mp.count(d)) return ((i*m-mp[d])%p+k+p)%p;
    }
    return -;
}

int main()
{
    while()
    {
        scanf("%lld%lld%lld",&x,&p,&y);
        if(!x&&!y&&!p) break;
        ll ans=exbsgs(x,y,p);
        if(ans==-) printf("No Solution\n");
        else printf("%lld\n",ans);
    }
    return ;
}