[BZOJ2440][中山市选2011]完全平方数（二分+容斥原理）

题目描述

传送门

题解

据说可以测出来对于一个数k它答案的上界最多时2*k。那么二分答案，每二分到一个mid，判断[1,mid]范围内有几个符合要求的数。
于是这个判定就用到了容斥原理。考虑计算有多少个数符合要求，也就是把[1,mid]的区间所有数的平方的倍数减掉。而这里的数如果是合数，那一定可以由另外的质数得到，所以只考虑质数就可以了。那么就是所有的数-一个质数的平方的倍数+2个质数乘积的平方的倍数-3个质数乘积的平方的倍数……，这里的质数两两不同。于是我们可以发现这里的“几个质数”和它的符号和莫比乌斯函数mu的关系。因为mu的一种定义方式就是：如果有平方数因数则为0，否则就是$(-1)^k$，k为约数个数。于是如果枚举到了一个数t，要计算[1,mid]的范围内有几个数是它的平方的倍数，这个结果显然是$\lfloor{mid \over {t^2}}\rfloor$。然后再乘上它的mu就得到了它对答案的贡献。因为是下取整，所以t枚举到$\sqrt {mid}$就可以了。
二分的时候要用LL，否则会炸掉。

数轮题仰仗ATP神犇啦啦啦啦，%%%%%%%

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
const int N=1e5;
int T,k;
LL ans;
int mu[N+5],p[N+5],prime[N+5];

inline void get_mu(){
    mu[1]=1; 
    for (int i=2;i<=N;++i){
        if (!p[i]){
            prime[++prime[0]]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=prime[0]&&i*prime[j]<=N;++j){
            p[i*prime[j]]=1;
            if (i%prime[j]==0){
                mu[i*prime[j]]=0;
                break;
            }
            else mu[i*prime[j]]=-mu[i];
        }
    }
}
inline int calc(int n){
    LL ans=0;
    for (int i=1;i*i<=n;++i)
      ans+=(LL)mu[i]*n/(i*i);
    return ans;
}
inline LL calc_ans(int k){
    LL l=0,r=(LL)2*k,mid,ans;
    while (l<=r){
        mid=(l+r)>>1;
        if (calc(mid)>=k) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main(){
    get_mu();
    scanf("%d",&T);
    while (T--){
        scanf("%d",&k);
        ans=calc_ans(k);
        printf("%lld\n",ans);
    }
}

总结

有很强的挫败感。