[BZOJ2301]Problem b 莫比乌斯反演+容斥

题意明确,就是求 i=abj=cdgcd(i,j)==k
首先容易发现容斥定理,转化为求
i=1nj=1mgcd(i,j)==k
我们设 f(d)=i=1nj=1m(gcd(i,j)==d)
F(d)=i=1nj=1m(d|gcd(i,j))
那么明显有 F(n)=n|df(d)
那么反演会有 f(i)=i|dμ(di)F(d)
又显然 F(d)=ndmd
时间复杂度一次查询 O(n)
考虑到 nd 只有 n 种取值,我们可以枚举取值,进行一下除法优化就可以了就优化为一次查询 O(n)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
bool vis[50001];
int mu[50001],prime[50001],n,a,b,c,d,k;
ll sum[50001];
void get_mobious()
{int i,j;
    memset(vis,0,sizeof(vis));
    mu[1]=1;
    int tot=0;
     for (i=2;i<=50000;i++)
      {
          if (vis[i]==0)
          {
              prime[++tot]=i;
              mu[i]=-1;
          }
          for (j=1;j<=tot;j++)
          {
              if (i*prime[j]>50000) break;
              vis[i*prime[j]]=1;
              if (i%prime[j]==0)
              {
                  mu[i*prime[j]]=0;
                  break;
            }
            else mu[i*prime[j]]=-mu[i];
          }
      }
    sum[0]=0;
    for (i=1;i<=50000;i++)
    sum[i]=sum[i-1]+mu[i];
}
ll cal(int x,int y)
{int r,i,pos;
ll s=0;
    r=min(x,y);
    for (i=1;i<=r;i=pos+1)
    {
        pos=min(x/(x/i),y/(y/i));
        s+=(sum[pos]-sum[i-1])*(x/i)*(y/i);
    }
    return s;
}
int main()
{
    cin>>n;
    get_mobious();
    while (n--)
    {
      scanf("%d%d%d%d%d",&a,&b,&c,&d,&k);
         ll ans1=cal(b/k,d/k);
         ll ans2=cal((a-1)/k,d/k);
         ll ans3=cal(b/k,(c-1)/k);
         ll ans4=cal((a-1)/k,(c-1)/k);
         printf("%lld\n",ans1-ans2-ans3+ans4);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值