码迷,mamicode.com
首页 > 其他好文 > 详细

[bzoj] 1101 Zap || 莫比乌斯反演

时间:2018-01-05 15:16:28      阅读:122      评论:0      收藏:0      [点我收藏+]

标签:前缀和   main   blog   char   利用   一个   source   ref   sum   

原题

求x<=a,y<=b,并且gcd(x,y)=d的x,y对数


\(\sum^{n}_{x=1}\sum^{m}_{y=1}[gcd(x,y)==d]\)
\(=\sum^{\lfloor \frac{n}{d} \rfloor}_{x=1}\sum^{\lfloor \frac{m}{d} \rfloor}_{y=1}[gcd(x,y)==1]\)
根据莫比乌斯反演,可以把式子化为
\(=\sum^{\lfloor \frac{n}{d} \rfloor}_{x=1}\sum^{\lfloor \frac{m}{d} \rfloor}_{y=1}\sum_{d|gcd(x,y)}\mu(d)\)
因为$d|gcd(x,y) $ => \(d|x,d|y\)
所以式子变为 \(\sum^{n/d}_{i=1}\mu(d)*\lfloor \frac{\lfloor \frac{n}{d} \rfloor}{d} \rfloor*\lfloor \frac{\lfloor \frac{m}{d} \rfloor}{d} \rfloor\)

用n/d的取值分一个段,然后对mu求一个前缀和即可解决……

分段:
因为我们知道\(\lfloor \frac{n}{d} \rfloor\)的取值有\(\sqrt n\)种取值,所以对于\(\lfloor \frac{n}{d} \rfloor\)\(\lfloor \frac{m}{d} \rfloor\)一共有\(\sqrt n + \sqrt m\)种情况
经典枚举方法:

for (int i=1,last=0;i<=a;i=last+1)
    {
        last=min(a/(a/i),b/(b/i));
        ans+=(ll)(a/i)*(b/i)*(sum[last]-sum[i-1]);
    }

对mu求前缀和:
利用欧拉筛法:

void init()
{
    miu[1]=sum[1]=1;
    for (int i=2;i<=N;i++)
    {
    if (!notprime[i]) miu[i]=-1,prime[++tot]=i;
    for (int j=1;j<=tot && i*prime[j]<=N;j++)
    {
        notprime[i*prime[j]]=1;
        if (i%prime[j]) miu[i*prime[j]]=-miu[i];
        else
        {
        miu[i*prime[j]]=0;
        break;
        }
    }
    sum[i]=sum[i-1]+miu[i];
    }
}

AC代码:

#include<cstdio>
#include<algorithm>
#define N 50000
typedef long long ll;
using namespace std;
int t,a,b,d,miu[N+10],prime[N+10],tot,sum[N];
bool notprime[N+10];
ll ans;

int read()
{
    int ans=0,fu=1;
    char j=getchar();
    for (;j<'0' || j>'9';j=getchar()) if (j=='-') fu=-1;
    for (;j>='0' && j<='9';j=getchar()) ans*=10,ans+=j-'0';
    return ans*fu;
}

void init()
{
    miu[1]=sum[1]=1;
    for (int i=2;i<=N;i++)
    {
    if (!notprime[i]) miu[i]=-1,prime[++tot]=i;
    for (int j=1;j<=tot && i*prime[j]<=N;j++)
    {
        notprime[i*prime[j]]=1;
        if (i%prime[j]) miu[i*prime[j]]=-miu[i];
        else
        {
        miu[i*prime[j]]=0;
        break;
        }
    }
    sum[i]=sum[i-1]+miu[i];
    }
}

int main()
{
    init();
    t=read();
    while (t--)
    {
    a=read();b=read();d=read();
    a/=d;
    b/=d;
    ans=0;
    if (a>b) swap(a,b);
    for (int i=1,last=0;i<=a;i=last+1)
    {
        last=min(a/(a/i),b/(b/i));
        ans+=(ll)(a/i)*(b/i)*(sum[last]-sum[i-1]);
    }
    printf("%lld\n",ans);
    }
    return 0;
}

[bzoj] 1101 Zap || 莫比乌斯反演

标签:前缀和   main   blog   char   利用   一个   source   ref   sum   

原文地址:https://www.cnblogs.com/mrha/p/8203612.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!