【LOJ】#3096. 「SNOI2019」数论

LOJ#3096. 「SNOI2019」数论

如果(P > Q)我们把(P)和(Q)换一下，现在默认(P < Q)

这个时候每个合法的(a_i)都可以直接落到(Q)中，因为(a_{i} equiv a_{i} pmod Q)这样避免了麻烦

然后呢我们发现每次把((a_{i} + P) \% Q)会走成一个圈，我们就要求从(a_{i})开始数(lfloor frac{T - 1- a_{i}}{P} floor + 1)个圈里(b_{i})的总和

这样的话可以断圈为链，复制两份就可以快速查某一段开始走小于圈长步的总和了

这样的圈一共有(gcd(P,Q))个

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('
')
#define eps 1e-10
#define ba 47
#define MAXN 1000005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
	if(c == '-') f = -1;
	c = getchar();
    }
    while(c >= '0' && c <= '9') {
	res = res * 10 +c - '0';
	c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
	out(x / 10);
    }
    putchar('0' + x % 10);
}
int P,Q,n,m;
int A[MAXN],B[MAXN];
int sum[MAXN * 2],pos[MAXN * 2],tot;
bool vis[MAXN];
int64 ans,T;
int gcd(int a,int b) {
    return b == 0 ? a : gcd(b,a % b);
}
void Solve() {
    read(P);read(Q);read(n);read(m);read(T);
    int x;
    for(int i = 1 ; i <= n ; ++i) {
	read(x);A[x] = 1;
    }
    for(int i = 1 ; i <= m ; ++i) {
	read(x);B[x] = 1;
    }
    if(P > Q) {
	swap(P,Q);
	for(int i = 0 ; i <= 1000000 ; ++i) swap(A[i],B[i]);
    }
    int g = gcd(P,Q);
    for(int i = 0 ; i < g ; ++i) {
	for(int j = 1 ; j <= tot ; ++j) sum[j] = 0;
	tot = 0;
	int x = i;
	while(!vis[x]) {
	    sum[tot + 1] = B[x] + sum[tot];
	    pos[tot + 1] = x;
	    ++tot;
	    vis[x] = 1;
	    x = (x + P) % Q;
	}
	int t = tot;
	for(int j = 1 ; j <= t ; ++j) {
	    if(pos[j] < P && A[pos[j]]) {
		if(T - 1 - pos[j] >= 0) {
		    int64 all = (T - 1 - pos[j]) / P + 1;
		    int64 cnt = all / t;
		    ans += cnt * sum[t];
		    int rem = all % t;
		    if(rem) {
			ans += sum[j + rem - 1] - sum[j - 1];
		    }
		}
	    }
	    sum[tot + 1] = sum[tot] + B[x];++tot;
	    x = (x + P) % Q;
	}
    }
    out(ans);enter;
}
int main(){
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
}