[BZOJ2301][HAOI2011]Problem b

BZOJ
Luogu
Description
对于给出的n个询问,每次求有多少个数对(x,y),满足a≤x≤b,c≤y≤d,且gcd(x,y) = k,gcd(x,y)函数为x和y的最大公约数。
Input
第一行一个整数n,接下来n行每行五个整数,分别表示a、b、c、d、k
Output
共n行,每行一个整数表示满足要求的数对(x,y)的个数
Sample Input
2
2 5 1 5 1
1 5 1 5 2
Sample Output
14
3
HINT
100%的数据满足:1≤n≤50000,1≤a≤b≤50000,1≤c≤d≤50000,1≤k≤50000

sol

如果做了那个题的话就会发现这两道题其实是一道题。
本题加入了下界的限制,但很显然我们做一个二维的容斥就好了。
别告诉我你不会二维容斥(就是二维前缀和那种搞法)

code

#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
const int N = 50005;
const int n = 50000;
int gi()
{
    int x=0,w=1;char ch=getchar();
    while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if (ch=='-') w=0,ch=getchar();
    while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return w?x:-x;
}
int mu[N],pri[N],tot,s[N];
bool zhi[N];
void Mobius()
{
    zhi[1]=true;mu[1]=1;
    for (int i=2;i<=n;i++)
    {
        if (!zhi[i]) pri[++tot]=i,mu[i]=-1;
        for (int j=1;j<=tot&&i*pri[j]<=n;j++)
        {
            zhi[i*pri[j]]=1;
            if (i%pri[j]) mu[i*pri[j]]=-mu[i];
            else {mu[i*pri[j]]=0;break;}
        }
    }
    for (int i=1;i<=n;i++) s[i]=s[i-1]+mu[i];
}
ll calc(int a,int b,int k)
{
    a/=k;b/=k;
    if (a>b) swap(a,b);
    int i=1,j;ll ans=0;
    while (i<=a)
    {
        j=min(a/(a/i),b/(b/i));
        ans+=1ll*(s[j]-s[i-1])*(a/i)*(b/i);
        i=j+1;
    }
    return ans;
}
int main()
{
    int T=gi();
    Mobius();
    while (T--)
    {
        int a=gi()-1,b=gi(),c=gi()-1,d=gi(),k=gi();
        printf("%lld\n",calc(b,d,k)-calc(a,d,k)-calc(b,c,k)+calc(a,c,k));
    }
    return 0;
}
posted @ 2018-01-04 09:56  租酥雨  阅读(...)  评论(...编辑  收藏