bzoj 2440 SQF - 莫比乌斯反演

二分答案后随便反演一下即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<climits>
#define N 100000
#define lint long long
using namespace std;
int p[N],mu[N],notp[N];
inline int prelude(int n)
{
    mu[1]=1;
    for(int i=2,c=0;i<=n;i++)
    {
        if(!notp[i]) p[++c]=i,mu[i]=-1;
        for(int j=1;j<=c&&(lint)p[j]*i<=n;j++)
        {
            int x=p[j]*i;notp[x]=1;
            if(i%p[j]) mu[x]=-mu[i];
            else { mu[x]=0;break; }
        }
    }
    return 0;
}
inline lint calc(lint n,lint ans=0)
{
    for(lint i=1;i*i<=n;i++)
        if(mu[i]) ans+=mu[i]*(n/(i*i));
    return ans;
}
int main()
{
    int T;scanf("%d",&T),prelude(N-1);
    while(T--)
    {
        int k;scanf("%d",&k);
        lint L=1,R=INT_MAX,mid=(L+R)>>1;
        while(L<=R)
        {
            if(calc(mid)>=k) R=mid-1;
            else L=mid+1;mid=(L+R)>>1;
        }
        printf("%lld\n",L);
    }
    return 0;
}