P4195 【模板】扩展BSGS

题目

题目

思路

先把exBSGS变成普通BSGS。
我们这样做：
$a^x\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}c)=>a^x+cy=b$
接下来我们不停地除a，c的最大公约数(d1)。
$b_1=\frac{a}{d_1}{a}^{x-1}+c_{1}y$
我们接下来接着除gcd(a,c₁)=d₂
$b_2=\frac{a^2}{d_1{d}_2}{a}^{x-2}+c_{2}y$
我们不难发现此时若b不被gcd整除，则无解。
设一共做了$n$次，$D=\frac{a^n}{d_1{d}_2\dots \dots d_n}$
剩下的$a^{x-n}\equiv b_n\cdot D^{-1}(\mathrm{m}\mathrm{o}\mathrm{d}{c}_n)$普通BSGS可解。

---

接下来考虑普通BSGS。
由指数模的周期性，得若有解，必有c以内的解。
我们考虑优（bi）美（tai）的分块。
分$\sqrt{n}$块，那么每块就是$m=\sqrt{n}$个数。
那么这时原式变成
$a^{im-j}\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}c)=>a^{im}\equiv a^{j}b(\mathrm{m}\mathrm{o}\mathrm{d}c)$
这里我们需要枚举$1<=i<=\sqrt{c}，0<=j<=\sqrt{c}$
我们把右式存入hash表里（也可以map，只要数据水），这里一共$O(\sqrt{n})$。
然后再枚举i，又$O(\sqrt{n})$。
code：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
//When I wrote this code,God and I unterstood what was I doing 
long long gcd(long long a,long long b)
{
	long long r=a%b;
	while (r)
	{
		a=b,b=r,r=a%b;
	}
	return b;
}
long long exgcd(long long a,long long b,long long &x,long long &y)
{
	if (b==0)
	{
		x=1,y=0;
		return a;
	}
	long long r=exgcd(b,a%b,y,x);
	y-=(long long)(a/b)*x;
	return r;
}
long long inv(long long a,long long m)
{
	long long x,y;
	exgcd(a,m,x,y);
	return (x%m+m)%m;
}
struct fP{
	long long tot,head[(((1<<16)-1)<<2)],nxt[500005],b[500005],v[500005],top,sck[500005];
	void init()
	{
		tot=0;
		while (top) head[sck[top--]]=0;
		return;
	}
	void insert(long long x,long long y)
	{
		long long hask=x%(((1<<16)-1)<<2);
		for (long long i=head[hask];i;i=nxt[i])
		{
			if (b[i]==x)
			{
				v[i]=y;
				return;
			}
		}
		if (!head[hask]) sck[++top]=hask;
		nxt[++tot]=head[hask],head[hask]=tot;
		b[tot]=x,v[tot]=y;
		return;
	}
	long long QuantAsk(long long x)
	{
		long long hask=x%(((1<<16)-1)<<2);
		for (long long i=head[hask];i;i=nxt[i])
		{
			if (b[i]==x) return v[i];
		}
		return -1;
	}
} kyx;
long long BSGS(long long a,long long b,long long c)
{
	if (b==1||c==1) return 0;
	long long cnt=0,Gcd=gcd(a,c),d=1;
	while (Gcd!=1)
	{
		if (b%Gcd!=0) return -1;
		cnt++,b/=Gcd,c/=Gcd;
		d=d*(a/Gcd)%c;
		Gcd=gcd(a,c);
	}
	b=b*inv(d,c)%c;
	kyx.init();
	long long block=ceil(sqrt(c)),p=1;
	for (long long i=0;i<block;i++)
	{
		if (p==b) return i+cnt;
		kyx.insert(p*b%c,i);
		p=p*a%c;
	}
	long long q=p,t;
	for (long long i=block;i-block+1<=c-1;i+=block)
	{
		t=kyx.QuantAsk(q);
		if (t!=-1) return i-t+cnt;
		q=q*p%c;
	}
	return -1;
}
int main()
{
	long long a,b,c;
	while (1)
	{
		scanf("%lld%lld%lld",&a,&b,&c);
		if (a==0&&b==0&&c==0) return 0;
		a%=b,c%=b;
		//if (check(a,b,c)) continue;
		long long wj=BSGS(a,c,b);
		if (wj==-1) puts("No Solution");
		else printf("%lld\n",wj);
	}
	return 0;
}
//Now,only God know
//但行好事，莫问前程