BZOJ 2440 完全平方数 (容斥＋莫比乌斯反演＋二分)

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

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

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2440

题目分析：题目要求第k个无平方因子数，我们显然不可能把答案都求出来再查询，这个数据范围首先想到的是二分，对于第1-n的无平方因子数我们可以用容斥定理得到，拿总的个数减去4的倍数(-n/4个)，减去9的倍数(-n/9个)，但是36既是4的倍数又是9的倍数，被减了两次，要加回来(+n/36)，这样容斥就出来了，前面的符号正好和数字开根号后对应的莫比乌斯函数相同，这样问题就简单了，还有一点要说明的是二分的上界开多大，这个也影响着莫比乌斯函数要开多大，我们不妨假设第k个无平方因子数不会超过2k，具体证明我也不会，但是最小的平方因子是4，也就是说每4个数里必然有一个是平方因子数，同时因为平方因子越往后越大，可以yy出平均每四个数有不超过两个平方因子数这个结论，所以第k个无平方因子数不会超过2k，(其实打表也可验证)，所以二分上界取2k+1即可，莫比乌斯函数开sqrt(2e9)差不多5e4的样子

1000ms过

#include 
#include 
#include 
#define ll long long
using namespace std;
int const MAX = 5e4;
ll const INF = 2e9;
int mob[MAX], p[MAX];
bool prime[MAX];

void Mobius()
{
    int pnum = 0;
    memset(prime, true, sizeof(prime));
    mob[1] = 1;
    for(int i = 2; i < MAX; i++)
    {
        if(prime[i])
        {
            p[pnum ++] = i;
            mob[i] = -1;
        }
        for(int j = 0; j < pnum && i * p[j] < MAX; j++)
        {
            prime[i * p[j]] = false;
            if(i % p[j] == 0)
            {
                mob[i * p[j]] = 0;
                break;
            }
            mob[i * p[j]] = -mob[i];
        }
    }
}

ll cal(int mid)
{   
    ll pos = 0;
    for(int i = 1; i * i <= mid; i++)
        pos += (ll) mob[i] * (mid / (i * i));
    return pos;
}

int main()
{
    Mobius();
    int T;
    scanf("%d", &T);
    while(T--)
    {
        ll k;
        scanf("%lld", &k);
        ll l = 1, r = 2 * k + 1;
        while(l <= r)
        {
            ll mid = (l + r) >> 1;
            if(cal(mid) < k)
                l = mid + 1;
            else
                r = mid - 1;
        }
        printf("%lld\n", l);
    }
}