bzoj2440 [中山市选2011]完全平方数

题目

首先，直接得出答案貌似不太可能，我们可以先二分一个答案，这样就成了计数问题了，就是求前n个数中，有多少个数没有平方因子。很显然的一点，就是$\mu$值不为0的数有多少个，但是，这样是算不出来的，因为太大了，所以我们就要想一想办法。考虑容斥原理，发现前面的系数刚好为$\mu$值，最后我们只需要算一个求和式子$\sum^{\sqrt{x}}_{i=1}[\frac{x}{i^2}]$就是算出来的答案了。

#include<bits/stdc++.h>
#define N 100000
using namespace std;
typedef long long ll;
int notp[N+5],prime[N+5],mu[N+5],cnt;
ll T,k,l,r,mid,ans;
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline ll read()
{
    ll x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void init()
{
    notp[1]=true,mu[1]=1;
    for(int i=2;i<=N;i++)
    {
        if(!notp[i])prime[++cnt]=i,mu[i]=-1;
        for(int j=1;j<=cnt&&i*prime[j]<=N;j++)
        {
            notp[i*prime[j]]=true;
            if(i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }
}
inline ll cal(ll x)
{
    ll rec=0;
    for(int i=1;i<=sqrt(x);i++)rec+=(ll)mu[i]*(x/((ll)i*i));
    return rec;
}
int main()
{
    freopen("in.txt","r",stdin);
    init();
    T=read();
    while(T--)
    {
        k=read(),l=1,r=k*10;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(cal(mid)>=k)ans=mid,r=mid-1;
            else l=mid+1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}