【校内模拟】游戏达人(BSGS求指标)(类欧几里得)

传送门


TM什么毒瘤省选模拟出类欧几里得。

题解:

由于本身对类欧几里得不是很熟,考场上也没有往类欧几里得方向去想。

结果你TM告诉我最后一步就是类欧几里得

首先特判无解和全零情况。

CxDy(modp)C^x\equiv D^y\pmod p看上去并不是很好操作,注意到pp是质数,也就是说pp有原根,那么我们直接求出C,DC,D的指标c,dc,d,则现在问题就是cxdy(modp1)cx\equiv dy\pmod {p-1}

首先,如果要解是很好解的,但是现在问题不是解这个方程,而是最小化Ax+ByAx+By。。。考场上没有什么时间去推,也就直接宣告凉凉。

下来滚去orz了题解的推导。

接下来是一系列非常牛逼的等价转换,设现在关于x,y的等价关系为cxdy(modp)cx\equiv dy \pmod p

首先同除以三者gcd,现在gcd(c,d,p)=1gcd(c,d,p)=1

g=gcd(p,c)g=gcd(p,c),显然gcd(d,g)=1gcd(d,g)=1,则gg一定是yy的因数,我们可以将BB乘上gg来进行等价变形。

得到的方程形式没有变,设上述变量表示现在的方程,现在设q=gcd(p,d)q=gcd(p,d),同理qq一定是xx的因数,则将AA乘上qq来等价。

现在得到方程cxdy(modp)cx\equiv dy\pmod p,最小化Ax+ByAx+By

由于我们能够保证现在gcd(c,p)=1gcd(c,p)=1,即cc%p\%p下有逆元。方程可以直接转化为xdy(modp)x\equiv dy\pmod p

考虑到x,yx,y都是正整数,我们需要最小化Ax+ByAx+By,且A,BA,B均为正数,显然我们确定了yy的时候x=dy%p=dypdypx=dy\%p=dy-p\lfloor\frac{dy}{p}\rfloorxx的最优选择,并且由于gcd(d,y)=1\gcd(d,y)=1,在y!=py!=p的时候xx一定非0,对于x=y=px=y=p的情况可以直接扔掉,显然不够优秀。

把求值的式子转化一下得到(Ad+B)yApdyp,y[1,p1](Ad+B)y-Ap\lfloor\frac{dy}{p}\rfloor,y\in[1,p-1]

现在要求的就是min{ay+bdypy[l,r]}\min\{ay+b\lfloor\frac{dy}{p}\rfloor\mid y\in[l,r]\}

神仙类欧几里得。

f(l,r,a,b,c,d,e)=min{ay+bcy+dey[l,r]}f(l,r,a,b,c,d,e)=\min\{ay+b\lfloor\frac{cy+d}{e}\rfloor\mid y\in[l,r]\},其中c,d0,e>0c,d\geq 0,e>0

然后是xjb推导,设p(y)=cy+de,L=p(l),R=p(r)p(y)=\lfloor\frac{cy+d}{e}\rfloor,L=p(l),R=p(r)

然后大力分类讨论

  1. L=RL=R根据AA的正负性,答案在两个端点取到,直接算一下就行了。
  2. cec\geq e,类欧几里得常规操作,f(l,r,a,b,c,d,e)=f(l,r,a+bce,b,c%e,d,e)f(l,r,a,b,c,d,e)=f(l,r,a+b\lfloor\frac{c}{e}\rfloor,b,c\%e,d,e)
  3. c<e,a0c< e,a\geq0,类欧几里得常规操作,此时显然有L<RL<R,由于c<ec<e,对于每个k[L,R]k\in[L,R],存在集合YkY_k,使得yYk,p(y)=k\forall y\in Y_k,p(y)=k,现在要最小化ay+bkay+bk,由于a0a\geq 0,所以现在显然考虑选择最小的yy,即kp(y)<kk\leq p(y)< k的最小yy。解不等式得到kedcy<ke+edc\frac{ke-d}{c}\leq y < \frac{ke+e-d}{c}
    转化成形式上一样的下取整,即ked1c<yke+ed1c\lfloor\frac{ke-d-1}{c}\rfloor<y\leq \lfloor\frac{ke+e-d-1}{c}\rfloor
    此时由于c<ec<e,总能够保证yy在该范围内有正整数取值,且最小的yyked+c1c\lfloor\frac{ke-d+c-1}{c}\rfloor。发现只有k=Lk=L的情况yy有可能取不到该最小值,直接计算即可。
    发现这个形式和上面的一模一样,问题转化为f(L+1,R,b,a,e,d+c1,c)f(L+1,R,b,a,e,-d+c-1,c)
  4. c<e,a<0c<e,a < 0,还是类欧几里得常规操作,yy尽量取最大值,不等式和上面一样,特判右端点,问题转化为f(L,R1,b,a,e,ed1,c)f(L,R-1,b,a,e,e-d-1,c)

然后就是找一下原根,然后BSGS求指标。

注意询问有点多,BSGS需要魔改块长。

然后我平生第一次知道了sqrt算负数会自动取unsigned值


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cout;
using std::cerr;

int p,gr;

inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=(ll)a*a%p)(b&1)&&(res=(ll)res*a%p);
	return res;
}

struct Map{
	static cs int magic=4898597;
	int key[magic],val[magic],waste;
	Map(){memset(key,-1,sizeof key);}
	int locate(int k)cs{
		int h=k%magic;
		while(key[h]!=-1&&key[h]!=k)h=(h+1)%magic;
		return h;
	}
	int &operator[](int k){
		int h=locate(k);
		if(key[h]==-1)key[h]=k;
		return val[h];
	}
	cs int &operator[](int k)cs{
		int h=locate(k);
		return key[h]==-1?waste:val[h];
	}
	bool find(int k)cs{return key[locate(k)]==k;}
}ma;

inline void get_gr(){
	int phi=p-1;
	std::vector<int> pr;
	for(int re i=2;(ll)i*i<=phi;++i){
		if(phi%i==0){
			pr.push_back(i);
			while(phi%i==0)phi/=i;
		}
	}
	if(phi>1)pr.push_back(phi);
	for(gr=2;;++gr){
		bool flag=true;
		for(int p:pr)if(power(gr,(::p-1)/p)==1){flag=false;break;}
		if(flag)break;
	}
}

int step,ibase;
inline void init(int T){
	get_gr();//cout<<"gr : "<<gr<<"\n";
	step=ceil(sqrt((ll)T*p));
	int now=1;
	for(int re i=0;i<step;++i){
		ma[now]=i;
		now=(ll)now*gr%p;
	}
	ibase=power(now,p-2);
}

inline int get_ind(int v){
	for(int re i=0;;++i,v=(ll)v*ibase%p){
		if(ma.find(v))
		return i*step+ma[v];
	}
	assert(0);
	return -1;
}

inline ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
inline void ex_gcd(ll a,ll b,ll &x,ll &y){
	if(!b){x=1,y=0;return ;}ex_gcd(b,a%b,y,x);y-=a/b*x; 
}
inline ll inv(ll a,ll b){
	ll x,y;ex_gcd(a,b,x,y);
	return (x%b+b)%b;
}

inline ll f(ll l,ll r,ll a,ll b,ll c,ll d,ll e){
	if(l>r)return 1ll<<60;
	ll L=(c*l+d)/e,R=(c*r+d)/e;
	if(L==R)return std::min(a*l,a*r)+L*b;
	if(c>=e)return f(l,r,a+b*(c/e),b,c%e,d,e);
	if(a>=0)return std::min(a*l+b*L,f(L+1,R,b,a,e,c-d-1,c));
	return std::min(a*r+b*R,f(L,R-1,b,a,e,e-d-1,c));
}

inline void solve(){
	ll a,b,c,d;
	scanf("%lld%lld%lld%lld",&a,&b,&c,&d);
	c%=p,d%=p;
	if((c==0)^(d==0)){cout<<"Damn it\n";return ;}
	if(c==0){cout<<a+b<<"\n";return ;}
	c=get_ind(c),d=get_ind(d);int p=::p-1;
	int g=gcd(gcd(c,d),p);
	c/=g,d/=g,p/=g;
	g=gcd(c,p);c/=g,p/=g,b*=g;
	d=(ll)d*inv(c,p)%p;
	g=gcd(d,p);d/=g,p/=g,a*=g;
	cout<<std::min((a+b)*p,f(1,p-1,a*d+b,-a*p,d,0,p))<<"\n";
}

signed main(){
#ifdef zxyoi
	freopen("game.in","r",stdin);
#endif
	int T;scanf("%d%d",&p,&T);init(T);
	while(T--)solve();
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章