【BZOJ5418】【NOI2018】屠龙勇士（数论，exgcd）

【NOI2018】屠龙勇士（数论，exgcd）

题面

洛谷

题解

考场上半个小时就会做了，一个小时就写完了。。
然后发现没过样例，结果大力调发现中间值爆\(longlong\)了，然后就没管了。。
然后又没切掉。。。我是真的傻逼。。。

首先每次选择的刀一定，直接一个\(multiset\)就算出来了。
然后对于每关都单独解一个方程
\(atk[i]x+p[i]y=a[i]\)，直接\(exgcd\)求解即可。
但是注意题目方程的含义，所以\(x\gt 0,y\le 0\)
所以要解出来之后还需要额外的计算一下（就是这里可能爆\(ll\)...)

那么此时对于每一个方程，我们都得到了一个最小的通解\(X0[i]\)
那么，一个可行解\(X=X0[i]+kd[i]\)，其中\(d[i]=p[i]/gcd(p[i],atk[i])\)，\(k\)是常数。
考虑如何合并两个解，
\(X0[1]+k_1d[1]=X0[2]+k_2d[2]\)
不妨令\(X0[2]\gt X0[1]\)，移项得
\(X0[2]-X0[1]=k_1d[1]+k_2d[2]\)
还是一个\(exgcd\)，同时\(k_1\ge 0,k_2\le 0\)，还是这里额外算一下，中间值可能爆\(ll\)
然后就可以算出这两个方程合并后的最小特解\(X0\)，
那么这两个方程合并后的通解就成了\(X=X0+lcm(d[1],d[2])\)
这样子顺次合并就行了。

至于中间值爆\(ll\)的问题，发现额外计算一下的过程就是一个取模+减法
所以龟速乘解决就好了。
然后无解就是某一步的时候\(exgcd\)无解，直接判就好。

为啥他们都说是拓展CRT，我怎么不知道啊？？？

这题我的代码写得好乱啊

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
using namespace std;
#define ll long long
#define MAX 100100
inline ll read()
{
	ll x=0;bool fl=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')fl=true,ch=getchar();
	while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();
	return fl?-x:x;
}
int n,m;
ll a[MAX],p[MAX],g[MAX],atk[MAX];
ll LCM(ll a,ll b){return (a/__gcd(a,b))*b;}
ll exgcd(ll a,ll b,ll &x,ll &y)
{
	if(b==0){x=1;y=0;return a;}
	ll d=exgcd(b,a%b,y,x);
	y-=a/b*x;return d;
}
namespace Choose
{
	multiset<ll> S;
	multiset<ll>::iterator it,itt;
	void Work()
	{
		for(int i=1;i<=n;++i)
		{
			it=itt=S.upper_bound(a[i]);
			if(it!=S.begin())--itt,atk[i]=*itt,S.erase(itt);
			else atk[i]=*it,S.erase(it);
			S.insert(g[i]);
		}
	}
}
ll X0[MAX],d[MAX];
void init()
{
	Choose::S.clear();
	memset(a,0,sizeof(a));memset(atk,0,sizeof(atk));
	memset(g,0,sizeof(g));memset(p,0,sizeof(p));
	memset(X0,0,sizeof(X0));memset(d,0,sizeof(d));
}
ll Multi(ll a,ll b,ll p)
{
	ll s=0;
	while(b){if(b&1)s=(s+a)%p;a=(a+a)%p;b>>=1;}
	return (s+p)%p;
}
bool Solve()
{
	ll x,y;
	for(int i=1;i<=n;++i)
	{
		ll D=__gcd(atk[i],p[i]),t,G,bs;
		if(a[i]%D)return false;
		exgcd(atk[i]/D,p[i]/D,x,y);
		G=p[i]/D;t=Multi(x,a[i]/D,G);
		if(t==0)t+=G;
		x=t;y=(a[i]-atk[i]*x)/p[i];
		if(y>0)
		{
			t=-y;G=atk[i]/D;
			t=(t%G+G)%G;bs=(t+y)/G;
			y=-t;x+=bs*(p[i]/D);
		}
		X0[i]=x,d[i]=p[i]/D;
	}
	for(int i=2;i<=n;++i)
	{
		if(X0[i]<X0[i-1])swap(X0[i],X0[i-1]),swap(d[i],d[i-1]);
		ll c=X0[i]-X0[i-1],D=__gcd(d[i],d[i-1]),G,t,bs;
		if(c%D!=0)return false;
		exgcd(d[i-1]/D,d[i]/D,x,y);
		G=d[i]/D;t=Multi(x,c/D,G);
		x=t;y=(c-x*d[i-1])/d[i];
		if(y>0)
		{
			t=-y;G=d[i-1]/D;
			t=(t%G+G)%G;bs=(t+y)/D;
			y=t;x+=bs*(d[i]/D);
		}
		X0[i]-=d[i]*y;d[i]=LCM(d[i],d[i-1]);
	}
	return true;
}
int main()
{
	freopen("dragon.in","r",stdin);
	freopen("dragon.out","w",stdout);
	int T=read();
	while(T--)
	{
		init();
		n=read();m=read();
		for(int i=1;i<=n;++i)a[i]=read();
		for(int i=1;i<=n;++i)p[i]=read();
		for(int i=1;i<=n;++i)g[i]=read();
		for(int i=1;i<=m;++i)Choose::S.insert(read());
		Choose::Work();
		if(!Solve())puts("-1");
		else printf("%lld\n",X0[n]);
	}
	return 0;
}