BZOJ 2440 完全平方数（莫比乌斯反演+容斥原理+二分）

Description
小X自幼就很喜欢数。但奇怪的是，他十分讨厌完全平方数。他觉得这些数看起来很令人难受。由此，他也讨厌所有是完全平方数的正整数倍的数。然而这丝毫不影响他对其他数的热爱。这天是小X的生日，小W想送一个数给他作为生日礼物。当然他不能送一个小X讨厌的数。他列出了所有小X不讨厌的数，然后选取了第 K个数送给了小X。小X很开心地收下了。 然而现在小W却记不起送给小X的是哪个数了。你能帮他一下吗？
Input
包含多组测试数据。文件第一行有一个整数T，表示测试数据的组数。
第2至第T+1行每行有一个整数Ki，描述一组数据，含义如题目中所描述。
(T<=50,1<=Ki<=10^9)
Output
含T 行，分别对每组数据作出回答。第 i 行输出相应的
第Ki 个不是完全平方数的正整数倍的数。
Sample Input
4
1
13
100
1234567
Sample Output
1
19
163
2030745
Solution

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 55555 
#define INF 2222222222
typedef long long ll;
bool check[maxn];
int prime[maxn],mu[maxn];
void Moblus(int n)
{
    memset(check,0,sizeof(check));
    mu[1]=1;
    int tot=0;
    for(int i=2;i<=n;i++)
    {
        if(!check[i])
        {
            prime[tot++]=i;
            mu[i]=-1;
        }
        for(int j=0;j<tot;j++)
        {
            if(i*prime[j]>n)break;
            check[i*prime[j]]=1;
            if(i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            else mu[i*prime[j]]=-mu[i];
        }
    }
}
ll count(ll x)
{
    ll ans=0;
    for(ll i=1;i*i<=x;i++)
        ans+=x/(i*i)*mu[i];
    return ans;
}
int main()
{
    Moblus(maxn-10);
    printf("%lld\n",count(100000000));
    int T,k;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&k);
        ll l=k,r=INF,ans;
        while(l<=r)
        {
            ll mid=(l+r)>>1;
            if(count(mid)>=k)ans=mid,r=mid-1;
            else l=mid+1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}