洛谷P2522 [HAOI2011]Problem b

洛谷P2522 [HAOI2011]Problem b

前置知识：莫比乌斯反演

题解

设$f(k)=\sum _{x=a}^b\sum _{y=c}^d[\gcd (x,y)=k],F(k)=\sum _{x=a}^b\sum _{x=c}^d[k|\gcd (x,y)]$

则有$F(k)=\sum _{k|n}f(n)$

由莫比乌斯反演可得

$f(k)=\sum _{k|n}\mu (\frac{n}{k})\times F(n)$

因为$F(k)=\sum _{x=a}^b\sum _{x=c}^d[k|\gcd (x,y)]=(\lfloor \frac{b}{k}\rfloor -\lfloor \frac{a-1}{k}\rfloor )\times (\lfloor \frac{d}{k}\rfloor -\lfloor \frac{c-1}{k}\rfloor )$

所以$f(k)=\sum _{k|n}\mu (\frac{n}{k})\times F(n)=\sum _{k|n}\mu (\frac{n}{k})\times (\lfloor \frac{b}{n}\rfloor -\lfloor \frac{a-1}{n}\rfloor )\times (\lfloor \frac{d}{n}\rfloor -\lfloor \frac{c-1}{n}\rfloor )$

令$T$为数据组数，$w$为$a,b,c,d$的最大值，则每次询问的时间复杂度为$O(\frac{w}{k})$，总时间复杂度为$O(T\times \frac{w}{k})$，一些数据可能过不了，所以我们考虑优化。

对于式子中的$(\lfloor \frac{b}{n}\rfloor -\lfloor \frac{a-1}{n}\rfloor )\times (\lfloor \frac{d}{n}\rfloor -\lfloor \frac{c-1}{n}\rfloor )$，我们可以用数论分块。对于一个左端点，取一个满足四个数都不变的右端点。总共能取$8\sqrt{w}$个值，于是每次询问的操作就优化到$O(\sqrt{w})$，总时间复杂度为$O(T\sqrt{w})$。

code

#include<bits/stdc++.h>
using namespace std;
const int N=200000;
int t,a,b,c,d,k,lst,z[N+5],p[N+5],mu[N+5],s[N+5];
long long ans;
int main()
{
	mu[1]=1;
	for(int i=2;i<=N;i++){
		if(!z[i]){
			p[++p[0]]=i;mu[i]=-1;
		}
		for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
			z[i*p[j]]=1;
			if(i%p[j]==0){
				mu[i*p[j]]=0;break;
			}
			mu[i*p[j]]=-mu[i];
		}
	}
	for(int i=1;i<=N;i++){
		s[i]=s[i-1]+mu[i];
	}
	scanf("%d",&t);
	while(t--){
		scanf("%d%d%d%d%d",&a,&b,&c,&d,&k);
		ans=0;lst=0;
		for(int i=k,vt;;i+=k){
			vt=N;
			if(a-1>=i) vt=min(vt,(a-1)/((a-1)/i));
			if(b>=i) vt=min(vt,b/(b/i));
			if(c-1>=i) vt=min(vt,(c-1)/((c-1)/i));
			if(d>=i) vt=min(vt,d/(d/i));
			if(vt>=N) break;
			i=vt/k*k;
			ans+=1ll*(s[i/k]-s[lst/k])*(b/i-(a-1)/i)*(d/i-(c-1)/i);
			lst=i; 
		}
		printf("%lld\n",ans);
	}
	return 0;
}