BZOJ_P2440 [中山市选2011]完全平方数(数论+莫比乌斯反演+容斥原理)

BZOJ传送门

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 2225 Solved: 1070
[Submit][Status][Discuss]
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

Source

Sol:
吃枣药丸，感觉自己已经傻了，啥都不会了
PoPoQQQ神犇的课件里有
最后的公式

#include<cstdio>
#include<cmath>
#define N 100005
int cnt,t;long long k;int pr[N],mu[N];bool b[N];
inline long long in(long long x=0,char ch=getchar()){while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x;}
void GetMu(){
    mu[1]=1;
    for(int i=2;i<N;i++){
        if(!b[i]) pr[++cnt]=i,mu[i]=-1;
        for(int j=1;j<=cnt&&i*pr[j]<N;j++){
            b[i*pr[j]]=1;if(!(i%pr[j])) break;
            mu[i*pr[j]]=-mu[i];
        }
    }
}
inline long long calc(long long x){
    long long lim=sqrt(x),res=0;
    for(int i=1;i<=lim;i++) res+=mu[i]*(x/(i*i));
    return res;
}
inline long long solve(long long x){
    long long l=0,r=x<<1,mid;
    while(l<r){
        mid=(l+r)>>1;
        if(calc(mid)>=x) r=mid;
        else l=mid+1;
    }return l;
}
int main(){
    GetMu();t=in();
    while(t--) k=in(),printf("%lld\n",solve(k));
    return 0;
}