题意:求第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;
}