bzoj2440 [中山市选2011]完全平方数 反演容斥+二分

题意：求第k个非完全平方数。
二分一下前mid中有多少个非完全平方数，判断用容斥做，通过反演得答案就是n-奇数个质数的平方的倍数的个数+偶数个质数的平方的倍数的个数
那么ans=Σμ[i]*(n/i^2) （i<=sqrt(n)）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
int n;
const int N=1e5+5;
int tot,pri[N],bz[N],mu[N];
inline ll cal(ll n)
{
    ll ans=0;
    fo(i,1,int(sqrt(n)))ans+=mu[i]*n/(i*i);
    return ans;
}
inline ll solve(int n)
{
    ll l=n,r=1e9,ans=0;
    while (l<=r)
    {
        ll mid=(l+r)>>1;
        if (cal(mid)>=n)r=mid-1,ans=mid;
        else l=mid+1;
    }
    return ans;
}
inline void pre()
{
    mu[1]=1;
    int tot=0;
    fo(i,2,N)
    {
        if (!bz[i])pri[++tot]=i,mu[i]=-1;
        fo(j,1,tot)
        {
            if (i*pri[j]>N)break;
            bz[i*pri[j]]=1;
            if (i%pri[j]==0)
            {
                mu[i*pri[j]]=0;
                break;
            }
            mu[i*pri[j]]=-mu[i];
        }
    }

}
int main()
{
    int cas;
    scanf("%d",&cas);
    pre();
    while (cas--)
    {
        scanf("%d",&n);
        printf("%lld\n",solve(n));
    }
    return 0;
}