[BZOJ1467][POJ3243]Clever Y（exBSGS+手写hash）

本文链接：https://blog.csdn.net/Blue_CuSO4/article/details/78993097

题目：

题解：

看来是模数不一定为质数的扩展BSGS了
$A^x≡B\ (mod\ C)$
扩展算法的原理就是通过提取公因数使得A,C互质，然后再用普通的BSGS求解。
首先要想使A,C互质，我们就要提取A和C的最大公约数。假设这个公约数是d，提取完了以后就变成了 $\frac{A}{d}A^{x−1}≡\frac{B}{d}(mod \frac{C}{d})$ 。如果B不能整除d，说明这个式子是无解的；那么现在新的 $C′$ 变成了 $\frac{C}{d}$ ，如果A和C′互质，就能直接用BSGS求解了，否则继续上面的过程，不断提取最大公因数。假设提取了t个最大公因数，它们的乘积为tmp，那么最后的式子就变成了 $\frac{A^t}{tmp}A^{x−t}≡\frac{B}{tmp}\ (mod \frac{C}{tmp})$ 。这个时候 $A$ 和 $\frac{C}{tmp}$ 是互质的，用逆元把tmp移到等式右边就可以用BSGS求解了。
【引自优秀的学姐】
诶为什么d=gcd(a,p)了a和p/d还是不互质啊，注意这里是要求a和p/d互质而不是a/d和p/d互质

注意如果b是1就直接返回0就好了

然后再BZOJ上顺利地A了，但是POJ上却T到哭，看来手写hash是必要的，其实就是用了链表前向星

代码：

#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long
using namespace std;
const int N=100003;
LL a,b,p,v[N],c[N];int nxt[N],point[N],tot; 
void addline(LL z,LL mb){int x=z%N; ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=z; c[tot]=mb;}
LL gcd(LL a,LL b){if (!b) return a;else return gcd(b,a%b);}
LL exgcd(LL a,LL b,LL &x,LL &y){if (!b) x=1,y=0;else exgcd(b,a%b,y,x),y-=(a/b)*x;}
LL find(LL mb)
{
    int x=mb%N;
    for (int i=point[x];i;i=nxt[i])//明明都固定到一个point了，为什么还要循环？因为point是%N之后的，很有可能有重复的很多元素 
      if (mb==v[i]) return c[i];
    return -1;
}
LL ksm(LL a,LL k)
{
    LL ans=1;
    for (;k;k>>=1,a=a*a%p)
      if (k&1) ans=ans*a%p;
    return ans;
}
LL bsgs(LL a,LL b,LL p)
{
    if (b==1) return 0;
    int k=0,t;LL xs=1;
    while ((t=gcd(a,p))!=1)
    {
        if (b%t) return -1;
        k++;p/=t;b/=t;xs=xs*(a/t)%p;
        if (b==t) return k;
    }
    LL lj,yy;exgcd(xs,p,lj,yy);
    b=b*lj;
    tot=0;    
    memset(point,0,sizeof(point));  
    LL m=ceil(sqrt(p));
    LL am=ksm(a,m);
    for (int j=0;j<=m;j++){addline(b,j); b=b*a%p;}
    LL mul=1;
    for (int i=1;i<=m;i++)
    {
        mul=mul*am%p;
        LL t=find(mul);
        if (t!=-1) return i*m-t+k;
    }
    return -1;
}
int main()
{
    while (scanf("%lld%lld%lld",&a,&p,&b))
    {
        if (!a && !b && !p) break; 
        LL ans=bsgs(a%p,b%p,p);
        if (ans==-1) printf("No Solution\n");
        else printf("%lld\n",ans);
    }
}