[CF1500B]Two chandeliers

题目

传送门 to CF

思路

第 $i=kn+a$ 天，考虑两个灯是否相同，就可以检查二分值。即
$$\sum _{kn+a\le t}[p_a\ne q_{(kn+a)\mathrm{m}\mathrm{o}\mathrm{d}m}]$$

设 $kn\mathrm{m}\mathrm{o}\mathrm{d}m=x_0$，实际上我们就要求
$$\sum _a[p_a\ne q_{(a+x_0)\mathrm{m}\mathrm{o}\mathrm{d}m}]$$

而 $(k+1)n\le t$ 时，$a$ 就没有限制了。所以多数时候 $a$ 是无限制的。如果对于所有 $x_0$ 都求出这个值，就不需要二分了，直接用 $k$ 暴力增，最后确定 $t-kn$ 的值。所以这个无限制的问题怎么做？

正难则反，考虑统计有多少个是相等的。那么 $p_a=q_b$ 会贡献到 $x_0=b-a$ 上面去。显然这就是卷积——将 $p$ 翻转即可。

问题是，下标的范围挺大的，而权值太多，就会出问题。考虑经典 “大小点”，设定阙值 $M$，当前权值数量不超过 $M$ 就平方硬算，否则 $m\log m$ 卷积。显然复杂度是
$$\mathcal{O}\left(mM+\frac{m^2\log m}{M}\right)\ge \mathcal{O}(m\sqrt{m\log m})$$

发现会 $TLE$，然后我就想不出来了。考试结束后，$\mathsf{H}\mathsf{a}\mathsf{n}\mathsf{d}\mathsf{I}\mathsf{n}\mathsf{D}\mathsf{e}\mathsf{v}\mathsf{i}\mathsf{l}$ 突然告诉我，$p,q$ 中数字互不相同，所以每种权值都是最多出现 $2$ 次……

所以就做到 $\mathcal{O}(n+m)$ 了。这不读题的毛病，啥时候能改改！

代码

#include <bits/stdc++.h>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
typedef long long int_;
inline int_ readint(){
	int_ a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(int_ x){
	if(x < 0) putchar('-'), x = -x;
	if(x > 9) writeint(x/10);
	putchar((x-x/10*10)^48);
}

const int MaxN = 1000005;
int a[MaxN], b[MaxN], cnt[MaxN];
int A[MaxN], B[MaxN];

int main(){
	int n = readint(), m = readint();
	int_ k = readint();
	int d = __gcd(n,m);
	rep(i,1,n) a[A[i] = readint()] = i;
	rep(i,1,m) b[B[i] = readint()] = i;
	if(n > m){
		swap(n,m), swap(a,b);
		swap(A,B); // for convenience
	}
	rep(i,1,m<<1) if(a[i] && b[i])
		++ cnt[(b[i]+m-a[i])%m];
	int_ peri = 0; rep(i,0,m/d-1)
		peri += n-cnt[1ll*i*n%m];
	int_ ans = 1ll*n*m/d*((k-1)/peri);
	k = (k-1)%peri+1; // period
	rep(i,0,m/d-1)
		if(k <= n-cnt[1ll*i*n%m]){
			int t = 1ll*i*n%m;
			for(int j=1; j<=n; ++j,++ans)
				if(A[j] != B[(j+t-1)%m+1])
				if((-- k) == 0) break;
			++ ans; break;
		}
		else{
			k -= (n-cnt[1ll*i*n%m]);
			ans += n; // n more days
		}
	printf("%lld\n",ans);
	return 0;
}