BZOJ2440:[中山市选2011]完全平方数(莫比乌斯函数)

Description

小 X 自幼就很喜欢数。但奇怪的是，他十分讨厌完全平方数。他觉得这些数看起来很令人难受。由此，他也讨厌所有是完全平方数的正整数倍的数。然而这丝毫不影响他对其他数的热爱。
这天是小X的生日，小 W 想送一个数给他作为生日礼物。当然他不能送一个小X讨厌的数。他列出了所有小X不讨厌的数，然后选取了第 K个数送给了小X。小X很开心地收下了。
然而现在小 W 却记不起送给小X的是哪个数了。你能帮他一下吗？

Input

包含多组测试数据。文件第一行有一个整数 T，表示测试数据的组数。
第2 至第T+1 行每行有一个整数Ki，描述一组数据，含义如题目中所描述。 

Output

含T 行，分别对每组数据作出回答。第 i 行输出相应的第Ki 个不是完全平方数的正整数倍的数。

Sample Input

4
1
13
100
1234567

Sample Output

1
19
163
2030745

HINT

对于 100%的数据有 1 ≤ Ki ≤ 10^9,T ≤ 50

Solution

PO姐讲的已经很明白了我为什么还要费劲写公式啊

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#define LL long long
#define N (100000+1000)
using namespace std;

LL T,x,k,vis[N],prime[N],mu[N],cnt;

void Get_mu()
{
    mu[1]=1;
    for (int i=2; i<=100000; ++i)
    {
        if (!vis[i]){prime[++cnt]=i; mu[i]=-1;}
        for (int j=1; j<=cnt && prime[j]*i<=100000; ++j)
        {
            vis[prime[j]*i]=true;
            if (i%prime[j]==0) break;
            mu[prime[j]*i]=-mu[i];
        }
    }
}

LL check(LL x)
{
    LL ans=0;
    for (LL i=1; i<=sqrt(x); ++i)
        ans+=x/(i*i)*mu[i];
    return ans;
}

int main()
{
    scanf("%lld",&T);
    Get_mu();
    while (T--)
    {
        scanf("%lld",&k);
        LL l=x,r=1e10,mid,now,ans;
        while (l<=r)
        {
            mid=(l+r)>>1; now=check(mid);
            if (now>=k) ans=mid,r=mid-1;
            else l=mid+1;
        }
        printf("%lld\n",ans);
    }
}