「Luogu2522」[HAOI2011]Problem b
Solution
题目要求
\[\sum_{x=a}^b\sum_{y=c}^d[gcd(x,y)=k]\]
设上式为\(Ans(a,b,c,d,k)\)
不妨来想一想\(a=1,c=1\)的时候怎么做
其实就跟「Luogu3455」[POI2007]ZAP-Queries一模一样了
略过一大堆式子,当\(a=c=1\)时
\[Ans(1,b,1,d,k)=\sum_{t=1}^{\frac{min(b,d)}{k}}\mu(t)\lfloor\frac{b}{kt}\rfloor\lfloor\frac{d}{kt}\rfloor\]
容斥一下可得
\[Ans(a,b,c,d,k)=Ans(1,b,1,d,k)-Ans(1,a-1,1,d,k)\\-Ans(1,b,1,c-1,k)+Ans(1,a-1,1,c-1,k)\]
愉快地调用就可以了
Code
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#define maxn 50005
#define N 50000
using namespace std;
typedef long long ll;
template <typename T> void read(T &t)
{
t=0;int f=0;char c=getchar();
while(!isdigit(c)){f|=c=='-';c=getchar();}
while(isdigit(c)){t=t*10+c-'0';c=getchar();}
if(f)t=-t;
}
int n;
int pri[maxn],pcnt,nop[maxn],mu[maxn];
int k;
void GetPrime()
{
mu[1]=1,nop[1]=1;
for(register int i=2;i<=N;++i)
{
if(!nop[i])pri[++pcnt]=i,mu[i]=-1;
for(register int j=1;j<=pcnt && i*pri[j]<=N;++j)
{
nop[i*pri[j]]=1;
if(i%pri[j]==0)break;
else mu[i*pri[j]]=-mu[i];
}
}
for(register int i=1;i<=N;++i)mu[i]+=mu[i-1];
}
int Calc(int a,int b)
{
int re=0,up=min(a,b)/k;
for(register int l=1,r;l<=up;l=r+1)
{
r=min(a/(a/l),b/(b/l));
re+=(mu[r]-mu[l-1])*(a/(l*k))*(b/(l*k));
}
return re;
}
int main()
{
GetPrime();
read(n);
while(n--)
{
int a,b,c,d;
read(a),read(b),read(c),read(d),read(k);
printf("%d\n",Calc(b,d)-Calc(b,c-1)-Calc(a-1,d)+Calc(a-1,c-1));
}
return 0;
}
原文地址:https://www.cnblogs.com/lizbaka/p/10513619.html
时间: 2024-10-09 09:22:06