bzoj 2440 -莫比乌斯函数的应用 + 容斥原理

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=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

 

可以看到k有点大，我们可以二分X，求【1，x】里有多少个非平方倍数的数

对于要判断的【1，x】，显然，对于质数2 3 5 7，他们的平方的倍数都是我们所不需要的，要减去【1，x】里他们倍数的个数，也就是X/（2*2），X/（i*i），

但是对于36，即使2^2又是3^3的倍数，我们减多了，要加回来，加回的是所有两个质数的倍数的个数，也就是X/(2*2*3*3),X/(3*3*5*5).....

很显然可以看出是容斥原理啦。

  也就是【1，x】的答案为x-   【1，x】里所有单个质数的平方的倍数 + 【1，x】里所有两个质数的平方的倍数 -  【1，x】里所有3个质数的平方的倍数.....

而且也可以知道由于要求的是平方的倍数，那么 最大的质数也就是sqrt(x)就可以啦。

复杂度显然  分解质因素sqrt(x) +容斥原理枚举 2^num，num为质数个数，显然不可接受。

进一步观察，可以发现，这个容斥里，都是奇数个质数的平方的倍数对应的系数为减，偶数个为加，这和莫比乌斯函数的定义一模一样：

关于莫比乌斯函数mu，他的定义如下：

因此 我们对 i=1:sqrt(x)  累加一遍，系数为mu(i)，便可得到答案。

跑一下发现1e9的答案为16xxxxxx（10位）

所以直接开inf到1e10就可以了

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long  ll;
const ll p =1000000007;
const int  N=100000;
bool is_prime[N+500];
int prime[N+50];
int mu[N+50];
ll tot;
void Moblus()
{
    tot = 0;
    mu[1] = 1;
    for(ll i = 2; i < N; i++)
    {
        if(!is_prime[i])
        {
            prime[tot++] = i;
            mu[i] = -1;
        }
        for(ll j = 0; j < tot && i*prime[j] < N; j++)
        {
            is_prime[i*prime[j]] = 1;
            if(i % prime[j])
            {
                mu[i*prime[j]] = -mu[i];
            }
            else
            {
                mu[i*prime[j]] = 0;
                break;
            }
        }
    }
}
ll ok(ll x)
{
    ll ret=0;
    for (ll i=1;i*i<=x;i++)
        ret+= x/(i*i)*mu[i];
    return ret;
}
int main()
{
    Moblus();
    int t;cin>>t;
    while(t--)
    {
        ll k;
        scanf("%lld",&k);
        ll l=1,r=1e10,ans=-1;
        while(l<=r)
        {
            ll mid=(l+r)>>1;
            if (ok(mid)>=k)
                r=mid-1,ans=mid;
            else l=mid+1;
        }
        printf("%lld\n",ans);
    }


    return 0;
}