[LOJ3096][数学]SNOI2019：数论

LOJ3096

实际上是要求一个方程$a_i+P{k}_1=b_j+Q{k}_2$，可以发现不断增加p的时候左边的取值$modQ$会成一个环，那就把这个环搞出来，把对应的所有b标上去，然后枚举a，先算一下最多走多少步，然后求个环上前缀和就完了

Code：

#include<bits/stdc++.h>
#define pb push_back
#define ll long long
using namespace std;
inline ll read(){
	ll res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=1e6+5;
ll t,ans,step[N];
int id[N],col[N],pos[N],a[N],b[N],w[N];
int len,p,q,n,m,sign=0;
vector<int>s[N],v[N];
int dfs(int x){
	if(id[x]) return 0;
	id[x]=sign;v[sign].pb(x);
	return col[x]+dfs((x+p)%q);
}
inline ll sum(int l,int x){return s[id[x]][pos[x]+l]-s[id[x]][pos[x]];}
int main(){
	p=read();q=read();n=read();m=read();t=read();
	for(int i=1;i<=n;i++) a[i]=read();
	for(int i=1;i<=m;i++) b[i]=read();
	if(p>q) swap(p,q),swap(n,m),swap(a,b);
	len=q/__gcd(p,q);
	for(int i=1;i<=m;i++) col[b[i]]=1;
	for(int i=1;i<=n;i++) step[i]=(t-1-a[i])/p;
	for(int i=0;i<q;i++) if(!id[i]) ++sign,w[sign]=dfs(i);
	for(int i=1;i<=sign;i++){
		for(int j=0;j<v[i].size();j++) pos[v[i][j]]=j;
		int L=v[i].size()-1;
		for(int j=0;j<L;j++) v[i].pb(v[i][j]);
		s[i].pb(col[v[i][0]]);
		for(int j=1;j<v[i].size();j++) s[i].pb(s[i][j-1]+col[v[i][j]]);
	}
	for(int i=1;i<=n;i++){
		ans+=1ll*(step[i]/len)*w[id[a[i]]];
		ans+=sum(step[i]%len,a[i])+col[a[i]];
	}
	cout<<ans;
	return 0;
}