[BZOJ3239][BSGS]Discrete Logging

题意

---

求$A^x\equiv B(mod~P)$的最小非负整数解

---

BSGS裸题

自己推一下：

令$x=im+j$ ，$m=\lceil\sqrt q\rceil$，

$A^{im}A^j\equiv B (mod ~P)$

$A^j\equiv B*inv(A^{im})(mod~P)$

已知$inv(1)=1$

由费马小定理得$inv(A^{im})=inv(A^{(i-1)m})*A^{p-1-m}~mod~P$（详细见代码后*）

那么inv就可以处理出来了。

把$A^{0...m}$hash存一下

再求$B*inv(A^{(1...m)m})$在hash表中查找。

#include <cstdio>
#include <map>
#include <iostream>
#include <cmath>

using namespace std;

typedef long long ll;

int p,b,n;
map<int,int> Mp;

inline int powf(ll x,int y,int p){
  int k=1; x%=p;
  while(y){
    if(y&1) k=k*x%p;
    x=x*x%p;
    y>>=1;
  }
  return k;
}

inline void solve(int y,int z,int p){
  y%=p;
  if(!y&&!z) {puts("1");return;}
  if(!y) {puts("no solution");return;}
  ll t=ceil(sqrt(p)),k=1,ine=1;
  Mp.clear(); Mp[1]=0;
  for(int i=1;i<t;i++){
    k=1ll*k*y%p;
    if(Mp.count(k)) continue;
    Mp[k]=i;
  }
  int tmp=powf(y,p-t-1,p);
  for(int i=0;i<t;i++){
    if(Mp.count(z*ine%p)) {printf("%d\n",i*t+Mp[z*ine%p]);return;}
    ine=1ll*ine*tmp%p;
  }
  puts("no solution");
}

int main(){
  while(~scanf("%d%d%d",&p,&b,&n)) solve(b,n,p);
}

*$inv(A^{im})=inv(A^{(i-1)m})*A^{p-1-m}\mod P$

$proof.$

$inv(A^{(i-1)m})=\frac{1}{A^{im-m}}$

$inv(A^{(i-1)m})*A^{p-1-m}=\frac{A^{p-1-m}}{A^{im-m}}$上下同乘$A^m$

$\frac{A^{p-1-m}}{A^{im-m}}=\frac{A^{p-1}}{A^{im}}$

又因为费马小定理$A^{p-1}\equiv 1(\mod p)$得

$\frac{A^{p-1}}{A^{im}} \mod p=\frac{1}{A^{im}}\mod p=inv(A^{im})$

所以$inv(A^{im})=inv(A^{(i-1)m})*A^{p-1-m}~mod~P$