【P2522 】[HAOI2011]Problem b 莫比乌斯反演

Tmotfl

于 2018-04-11 21:15:00 发布

阅读量200

点赞数

本文链接：https://blog.csdn.net/NLt_XXy/article/details/79903188

版权

组合数学同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

莫比乌斯反演

2 篇文章 0 订阅

订阅专栏

题目描述

对于给出的n个询问，每次求有多少个数对(x,y)，满足a≤x≤b，c≤y≤d，且gcd(x,y) = k，gcd(x,y)函数为x和y的最大公约数。

输入输出格式

输入格式：

第一行一个整数n，接下来n行每行五个整数，分别表示a、b、c、d、k

输出格式：

共n行，每行一个整数表示满足要求的数对(x,y)的个数

输入输出样例

输入样例#1：

2
2 5 1 5 1
1 5 1 5 2

输出样例#1：

14
3

题解

对于题目中要求，我们可以进行容斥原理
来把问题转化为求解 $\sum\limits_{i=1}^{n}\sum\limits_{j=1}^{m}gcd(i,j)=k$ 的情况

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#define ll long long 
#define N 50020
const int MAXN=1e6;
using namespace std;
int pre[MAXN],vis[MAXN],mu[MAXN],S_mu[MAXN],num;
ll a,b,c,d,k;
void Init()
{
    mu[1]=1;
    for(int i=2;i<=N;i++)
    {
        if(vis[i]==0) pre[++num]=i,mu[i]=-1;
        for(int j=1;j<=num&&pre[j]*i<=N;j++)
        {
            vis[pre[j]*i]=1;
            if(i%pre[j]==0)
            {
                mu[pre[j]*i]=0;
                break;
            }
            else
                mu[pre[j]*i]=-mu[i];
        }
    }
    for(int i=1;i<=N;i++) S_mu[i]=S_mu[i-1]+mu[i];
}
int T;
ll Get(ll n,ll m)
{
    n/=k,m/=k;
    ll ans=0;
    for(ll i=1,j=0;i<=min(n,m);i=j+1)
    {
        j=min(n/(n/i),m/(m/i));
        ans+=(S_mu[j]-S_mu[i-1])*(n/i)*(m/i);
    }
    return ans;
}
int main()
{
    Init();
    scanf("%d",&T);
    while(T--)
    {
        ll ans=0;
        scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&k);   
        printf("%lld\n",(Get(b,d)-Get(a-1,d)-Get(b,c-1)+Get(a-1,c-1)));//容斥
    }
    return 0;
}