BZOJ 2440: [中山市选2011]完全平方数 莫比乌斯 容斥原理 二分

2440: [中山市选2011]完全平方数

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 3700  Solved: 1814
[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

月初写的

今天再看到竟然不会。。。GG

决定重学莫比乌斯（原来也没写几个题 笑哭）

求第k个无平方因子数

应该会立刻想到mobius函数吧

然后。。就GG了

那正解是什么呢

二分答案

如何判定呢

容斥原理

符合条件的数字=全部数字-一个质因子平方的倍数的数量+两个质因子平方的倍数的数量+····

发现仍然不好搞

但是 我们有莫比乌斯函数

容斥式子里的符号不就是对应的模拟乌斯函数值吗

所以就可以O(根号n。。。)判定了

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}
const int N=50000;
int mobius[N],prime[N>>3],cnt;
bool book[N];
void initial()
{
    mobius[1]=1;
    for(int i=2;i<N;i++)
    {
        if(!book[i]){prime[++cnt]=i;mobius[i]=-1;}
        for(int j=1;j<=cnt&&prime[j]*i<N;j++)
        {
            int now=prime[j]*i;book[now]=1;
            if(i%prime[j])mobius[now]=-mobius[i];
            else {mobius[now]=0;break;}
        }
    }
}
inline ll check(ll x)
{//cout<<x<<endl;
    ll sum=0;
    for(ll i=1;i*i<=x;i++)
    {sum+=x/(i*i)*mobius[i];}//cout<<sum<<endl;
    return sum;
}
int main()
{
    initial();
    //for(int i=1;i<=100;i++)cout<<mobius[i]<<" "; 
    int T=int(read());
    while(T--)
    {
        ll k=read();
        ll l=0,r=k<<1;
        while(l<r)
        {
            ll mid=l+r>>1;
            ll tmp=check(mid);
            if(tmp<k)l=mid+1;
            else r=mid;
        }
        printf("%lld\n",r);
    }
    return 0;
}
/*
4 
1 
13 
100 
1234567 
 
1 
19 
163 
2030745 
*/