莫比乌斯反演 bzoj 2301: [HAOI2011]Problem b模板

最新推荐文章于 2020-02-22 14:50:15 发布

Jacky35

最新推荐文章于 2020-02-22 14:50:15 发布

阅读量378

点赞数

分类专栏：模板库莫比乌斯反演模板库

本文链接：https://blog.csdn.net/u011056504/article/details/51746996

版权

模板库同时被 3 个专栏收录

43 篇文章 0 订阅

订阅专栏

模板库

39 篇文章 6 订阅

订阅专栏

莫比乌斯反演

5 篇文章 0 订阅

订阅专栏

求 $\mu$

mu[1]=1;
    fo(i,2,N)
    {
        if (bz[i]==0) p[++p[0]]=i,mu[i]=-1;
        fo(j,1,p[0])
        {
            int x=i*p[j];
            if (x>N) break;
            bz[x]=1;
            if (i % p[j]==0) {mu[x]=0;break;}
            mu[x]=-mu[i];
        }
    }

总

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 51000
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
int a,b,c,d,k,mu[N],p[N],bz[N],f[N],g[N],s[N];
int main()
{
    int ac;scanf("%d",&ac);
    mu[1]=1;
    fo(i,2,N)
    {
        if (bz[i]==0) p[++p[0]]=i,mu[i]=-1;
        fo(j,1,p[0])
        {
            int x=i*p[j];
            if (x>N) break;
            bz[x]=1;
            if (i % p[j]==0) {mu[x]=0;break;}
            mu[x]=-mu[i];
        }
    }
    s[0]=0;fo(i,1,N) s[i]=s[i-1]+mu[i];
    for(;ac;ac--)
    {
        scanf("%d%d%d%d%d",&a,&b,&c,&d,&k);
        int n=b,m=d;
        if (n>m) swap(n,m);
        f[k]=0;
        for(int j=1,i=1;i<=n/k;i=j+1) 
        {
            j=min(n/(n/i),m/(m/i));
            f[k]+=(s[j]-s[i-1])*(n/(i*k))*(m/(i*k));
        }
        n=a-1;m=d; if (n>m) swap(n,m);
        for(int j=1,i=1;i<=n/k;i=j+1) 
        {
            j=min(n/(n/i),m/(m/i));
            f[k]-=(s[j]-s[i-1])*(n/(i*k))*(m/(i*k));
        }
        n=b;m=c-1; if (n>m) swap(n,m);
        for(int j=1,i=1;i<=n/k;i=j+1) 
        {
            j=min(n/(n/i),m/(m/i));
            f[k]-=(s[j]-s[i-1])*(n/(i*k))*(m/(i*k));
        }
        n=a-1;m=c-1; if (n>m) swap(n,m);
        for(int j=1,i=1;i<=n/k;i=j+1) 
        {
            j=min(n/(n/i),m/(m/i));
            f[k]+=(s[j]-s[i-1])*(n/(i*k))*(m/(i*k));
        }
        printf("%d\n",f[k]);
    }
}