[二分+容斥+莫比乌斯函数] BZOJ2440: [中山市选2011]完全平方数

最新推荐文章于 2021-07-13 16:54:35 发布

Lynstery

最新推荐文章于 2021-07-13 16:54:35 发布

阅读量439

点赞数

分类专栏：莫比乌斯反演 --------------数学相关------------ 容斥 --------------其他--------------

本文链接：https://blog.csdn.net/CHHNZ/article/details/74129978

版权

102 篇文章 0 订阅

订阅专栏

--------------其他--------------

97 篇文章 0 订阅

订阅专栏

莫比乌斯反演

14 篇文章 0 订阅

订阅专栏

题意

求从小到大第k个无平方因子数是多少。（原题面表述略有问题）
k<=1e+9

题解

这题与反演无关，就用到了莫比乌斯函数。
首先肯定二分答案然后验证，现在问题转化为了求[1,mid]中无平方因子数的个数。
怎么求呢？考虑容斥，（p是1~sqrt(m)内的质数集）
$总个数=[(p_i)^2的倍数]-[(p_i*p_j)^2的倍数]+[(p_i*p_j*p_k)^2的倍数]-[(p_i*p_j*p_k*p_t)^2的倍数]......$
怎么求和呢？通过观察我们发现，对于任意 $[x^2的倍数]$ 这一项的前面的系数正好是mu[x]啊。
然后就完美解决了，
$ans_{1到m无平方因子数的个数}=\sum_i^{\lfloor \sqrt m \rfloor}\mu(i)*\lfloor m/i^2\rfloor$

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100015, N=100000;
int _test,n,m,mu[maxn],p[maxn];
LL ans;
bool vis[maxn];
void get_mu(){
    memset(vis,1,sizeof(vis));
    mu[1]=1;
    for(int i=2;i<=N;i++){
        if(vis[i]) p[++p[0]]=i, mu[i]=-1;
        for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
            vis[i*p[j]]=false;
            if(i%p[j]==0){ mu[i*p[j]]=0; break; }
            mu[i*p[j]]=-mu[i];
        }
    }
}
LL get(LL mid){
    LL res=0;
    int t=sqrt(mid);
    for(LL i=1;i<=t;i++) res+=(mid/(i*i))*mu[i];
    return res;
}
int main(){
    freopen("bzoj2440.in","r",stdin);
    freopen("bzoj2440.out","w",stdout);
    get_mu();
    scanf("%d",&_test);
    while(_test--){
        scanf("%d",&m);
        LL L=m,R=1e+10; ans=1;
        while(L<=R){
            LL mid=(L+R)>>1;
            if(get(mid)>=m) R=mid-1, ans=mid;
                       else L=mid+1; 
        }
        printf("%lld\n",ans);   
    }
    return 0;

}