【二分+容斥+莫比乌斯反演】BZOJ2440 完全平方数

 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<cstring>
 4 #include<cmath>
 5 using namespace std;
 6 const int maxn=5e4+5;
 7 
 8 int mu[maxn],flag[maxn];
 9 int prime[maxn],cnt;
10 
11 int getmu(){
12     mu[1]=1;
13     for(int i=2;i<maxn;i++){
14         if(!flag[i]){
15             prime[++cnt]=i;
16             mu[i]=-1;
17         }
18         for(int j=1;i*prime[j]<maxn&&j<=cnt;j++){
19             flag[i*prime[j]]=1;
20             if(i%prime[j]==0){
21                 mu[i*prime[j]]=0;
22                 break;
23             }
24             mu[i*prime[j]]=-mu[i];
25         }
26     }
27 }
28 
29 int work(int n){
30     int ret=0;
31     for(int i=1;i*i<=n;i++)
32         ret+=mu[i]*int(1.0*n/(i*i));
33     return ret;
34 }
35 
36 int main(){
37     int T,k;
38     scanf("%d",&T);
39     getmu();
40     
41     while(T--){
42         scanf("%d",&k);
43         int l=1,r=2*k;
44         while(l<r){
45             int mid=(l>>1)+(r>>1)+(l&r&1);
46             if(work(mid)>=k) r=mid;
47             else l=mid+1;
48         }
49         printf("%d\n",l);
50     }
51     return 0;
52 }