BZOJ 2440 完全平方数 （二分 莫比乌斯容斥）

BZOJ 2440 完全平方数

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

思路：
考虑二分答案。
check判断当前二分的ans比k大或小。
对于一个数t，t以内的数里的非完全平方数倍数的个数
num=1的倍数的数量−一个质数平方数(9,25,49…)的倍数的数量+两个质数的积平方数(36,100,225…)的数量−三个质数balabala。
所以正好可以用莫比乌斯函数处理。

#include <cstdio>
#include <algorithm>
#define N 100010
using namespace std;

int t,k,isnot[N],mu[N],prime[N],ptot;

void pre(){
    mu[1]=1;
    for(int i=2;i<N;i++){
        if(!isnot[i])
            prime[++ptot]=i,mu[i]=-1;
        for(int j=1;j<=ptot&&1LL*i*prime[j]<N;j++){
            isnot[i*prime[j]]=1;
            mu[i*prime[j]]=-mu[i];
            if(i%prime[j]==0){
                mu[i*prime[j]]=0;
                break;
            }
        }
    }
}

int check(int mid){
    int rt=0;
    for(int i=1;i*i<=mid;i++)
    rt+=mid/(i*i)*mu[i];
    return rt;
}

int solv(){
    int l=1,r=k<<1;  
    while(l+1<r)  
    {  
        int mid=(l>>1)+(r>>1)+(l&r&1);  
        if( check(mid)>=k )  
            r=mid;  
        else 
            l=mid;  
    }  
    if( check(l)>=k )  
        return l;  
    return r; 
}

int main(){
    pre();
    scanf("%d",&t);
    while(t--){
        scanf("%d",&k);
        printf("%d\n",solv());
    }
}