小 X 自幼就很喜欢数。但奇怪的是,他十分讨厌完全平方数。他觉得这些
数看起来很令人难受。由此,他也讨厌所有是完全平方数的正整数倍的数。然而
这丝毫不影响他对其他数的热爱。
这天是小X的生日,小 W 想送一个数给他作为生日礼物。当然他不能送一
个小X讨厌的数。他列出了所有小X不讨厌的数,然后选取了第 K个数送给了
小X。小X很开心地收下了。
然而现在小 W 却记不起送给小X的是哪个数了。你能帮他一下吗?
首先求第K个非完全平方数,一定可以在[1,2k]的范围内找到第k个非完全平方数!!!!
然后我首先想到的是容斥原理,我先求[1,2k]内完全平方数的个数,
即4的倍数的,9的倍数,16的倍数···然后奇加偶减,再用[1,2k]总的数的个数即2k减去就是[1,2k]内非完全平方数的个数,但是这样不好写代码.
看题解知道应用莫比乌斯函数来求解,大多题解都说是莫比乌斯函数+容斥定理,其实没有用到容斥,只是用了莫比乌斯函数;
核心代码
bool judge(ll x)
{
ll i,ans=0;
for(i=1;i*i<=x;i++)
ans+=mu[i]*x/(i*i);
if(ans<k)
return 1;
return 0;
}
i=1时就是[1,x]区间数的总数,i=2时,把四的倍数筛掉了,i=3把9的倍数筛掉了,i=4时mu[i]=0这不是没算?
因为i=2时就把i=4的倍数筛掉了,继续下去;
然后就是二分的应用!!!!
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<iomanip>
#include<vector>
#include<map>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=100000;
ll mu[maxn+5],prim[maxn+5],cnt,k;
bool v[maxn+5];
void init()
{
memset(v,0,sizeof(v));
mu[1]=1;
cnt=0;
for(int i=2;i<maxn+1;i++)
{
if(!v[i])
{
prim[cnt++]=i;
mu[i]=-1;
}
for(int j=0;j<cnt&&i*prim[j]<maxn+1;j++)
{
v[i*prim[j]]=1;
if(i%prim[j])
mu[i*prim[j]]=-mu[i];
else
{
mu[i*prim[j]]=0;
break;
}
}
}
}
bool judge(ll x)
{
ll i,ans=0;
for(i=1;i*i<=x;i++)
ans+=mu[i]*x/(i*i);
if(ans<k)
return 1;
return 0;
}
int main()
{
ll t,l,r,mid;
scanf("%lld",&t);
init();
while(t--)
{
scanf("%lld",&k);
l=1;r=(k<<1);
while(l<r)
{
mid=(l+r)/2;//cout<<1<<endl;
if(judge(mid))
l=mid+1;
else
r=mid;
}
//cout<<l<<" "<<r<<endl;
cout<<l<<endl;
}
}