【BZOJ2440】完全平方数

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 $\le$Ki $\le$ 10$^9$

, T $\le$ 50

Source
突然想起来了就去刷了个数论。
其实这是我在某个莫比乌斯反演的PPT里看到的，但是这个题不是反演只是个莫比乌斯函数的应用。
具体做法是二分答案。
只需要一个小小的check函数来判断当前二分到的答案是否比k大或小即可。
手动模拟了一下发现某个规律
对于一个数t，t以内的数里的非完全平方数倍数的个数
$num=1的倍数的数量-一个质数平方数(9,25,49...)的倍数的数量+两个质数的积平方数(36,100,225...)的数量-三个质数balabala$
所以上面那个规律正好可以用莫比乌斯函数来搞。
比如$\mu(3)=-1,\mu(6)=1...$
所以答案$ans=\sum_{i=1}^\sqrt x \mu(i)\lfloor \frac x {i^2}\rfloor$
二分时候出了点奇怪的问题。。。（这个题的二分姿势有点奇怪而且l+r会爆int）所以最后去膜拜了一下那个PPT原作者PoPoQQQ的代码拿来对拍。。。直到最后发现了问题改了过来$=\omega=$（别骂我QAQ）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 100000
using namespace std;
int T;
int k;
bool not_prime[MAXN];
int prime[MAXN];
int num;
int mu[MAXN];
void choose_prime()
{
    for (int i=2;i<=MAXN;i++)
    {
        if (!not_prime[i])
            prime[++num]=i;
        for (int j=1;j<=num&&i*prime[j]<=MAXN;j++)
        {
            not_prime[i*prime[j]]=1;
            if (i%prime[j]==0) break;
        }   
    }
}
void mob()
{
    for (int i=2;i<=MAXN;i++)
    {
        if (!not_prime[i]) mu[i]=-1;
        for (int j=1;j<=num&&i*prime[j]<=MAXN;j++)
        {
            if (i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            else
                mu[i*prime[j]]=-mu[i];
        }
    }
}
int check(long long t)
{
    long long ret=0;
    for (long long i=1;i*i<=t;i++)
        ret+=t/(i*i)*mu[i];
    return ret;
}
int main()
{
    scanf("%d",&T);
    mu[0]=0;mu[1]=1;
    choose_prime();mob();
    while (T--)
    {
        scanf("%d",&k);
        int l=1,r=k<<1;
        int ans;
        while (l+1<r)
        {
            int mid=(l>>1)+(r>>1)+(l&r&1);
            int temp=check(mid);
            if(temp>k) r=mid;
            else
            if (temp<k) l=mid;
            else 
            {
                ans=mid;
                r=mid;
            }
        }
        if (k==1) cout<<1<<endl;
        else
        printf("%d\n",ans);
    }
}