洛谷OJ：P4195 【模板】扩展BSGS（扩展BSGS）

思路：和BSGS模板题类似，只是这道题没有规定a与p互质，因此朴素的BSGS算法已经无法解决该题，于是引入了扩展BSGS算法，其实该算法不难，既然a与p不互质，呢我们经过变形将其变为互质之后就可以继续使用BSGS算法了，截取一段网上的证明：截图出处。

#include<set>
#include<map>
#include<queue>
#include<vector>
#include<string>
#include<math.h>
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
#define maxn 100050
#define SQ 1005
#define ll long long
ll p,n,b;
ll q(ll x,ll y,ll m){
	ll res=1;
	x=x%m;
	while(y){
		if(y&1)
			res=res*x%m;
		x=x*x%m;
		y/=2;
	}
	return res;
}
ll gcd(ll a,ll b){
	if(b==0)
		return a;
	return gcd(b,a%b);
}
ll phi(ll n){
	ll res=n;
	for(int i=2;i*i<=n;i++){
		if(n%i==0)
			res=res/i*(i-1);
		while(n%i==0)
			n/=i;
	}
	if(n>1)
		res=res/n*(n-1);
	return res;
}
ll BSGS(ll a,ll b,ll m,ll k=1){
	map<ll,ll> hs;
	ll cur=b*a%m,t=sqrt(m)+1;
	for(int B=1;B<=t;++B){
		hs[cur]=B;
		cur=cur*a%m;
	}
	ll at=q(a,t,m),now=at*k%m;
	for(int A=1;A<=t;A++){
		if(hs[now])
			return A*t-hs[now];
		now=now*at%m;
	}
	//这里返回一个稍微小一点的负数，确保多次加一后仍然是负数（表示无解）
	return -1000;
}
ll exBSGS(ll a,ll b,ll m,ll k=1){
	if(b==1)
		return 0;
	if(a==0 && b==0)
		return 1;
	ll d=gcd(a,m);
	if(b%d)
		return -1;
	else if(d==1)
		return BSGS(a,b,m,k%m);
	else
		return exBSGS(a,b/d,m/d,k*a/d%m)+1;
}
int main(void){
	while(scanf("%lld%lld%lld",&b,&p,&n),p+b+n>0){
		ll phiP=phi(p);
		ll ans=exBSGS(b%p,n%p,p);
		if(ans>phiP) ans=ans%phiP+phiP; //扩展欧几里得
		if(ans<0)
			printf("No Solution\n");
		else
			printf("%lld\n",ans);
	}
	return 0;
}