bzoj2301 [HAOI2011]Problem b

题目

莫比乌斯第一题。。。

还不是很熟悉，当时半推公式半找规律的。。
233
数学题，就是要多练习啊。

#include<bits/stdc++.h>
#define MAXN 50000
#define LL long long
using namespace std;
int prime[MAXN+1],size=0,mu[MAXN+1];
int sum_mu[MAXN+1];
bool P[MAXN+1];
int a,b,c,d,n,m,k,T;
void init()
{
    memset(P,true,sizeof(P));
    memset(mu,0,sizeof(mu));
    size=0;
    P[1]=false;
    mu[1]=1;
    for(int i=2;i<=MAXN;i++)
    {
        if(P[i])
        {
            prime[++size]=i;
            mu[i]=-1;
        }
        for(int j=1;j<=size&&i*prime[j]<=MAXN;j++)
        {
            P[i*prime[j]]=false;
            if(!(i%prime[j]))
            {
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }
    sum_mu[0]=0;
    for(int i=1;i<=MAXN;i++)
        sum_mu[i]=sum_mu[i-1]+mu[i];
}
LL G(int n,int m,int k)
{
    if(n>m)swap(n,m);
    LL ans=0;
    n=n/k;m=m/k;
    int i=1,j=0;
    while(i<=n)
    {
        j=min(n/(n/i),m/(m/i));
        ans+=(LL)(sum_mu[j]-sum_mu[i-1])*(n/i)*(m/i);
        i=j+1;
    }
    return ans;
}
int main()
{
    init();
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d%d%d",&a,&b,&c,&d,&k);
        printf("%d\n",G(b,d,k)+G(a-1,c-1,k)-G(a-1,d,k)-G(c-1,b,k));
    }
    return 0;
}