【BZOJ2440】[中山市选2011]完全平方数

【题目链接】

• BZOJ2440

【前置技能】

• 反演
• 容斥原理
• 二分答案

【题解】

• 题目要求的是第$k$个$\mu$值不为$0$的数，$k$达到了$10^9$，显然不能把$\mu$全部筛出来直接暴力找。
• 考虑二分答案，那么问题转化为求 $1\sim n$中的$\mu$值为零的数的个数。容斥原理：个数为所有素数的平方的倍数的个数；发现两个素数乘积的平方的倍数被重复计算(如$k\ast 2^2\ast 3^2$，会被$k\ast 2^2$和$k\ast 3^2$分别计算一次)，将其减去；发现三个素数乘积的平方的倍数被重复计算，将其加上……发现容斥系数就是$\mu$的值，所以$1$~$n$中的$\mu$值为零的数的个数为$\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }\mu (i)\ast \frac{n}{i^2}$。
• 二分的最大值应为$2\ast k$，$2^2$的倍数每$4$个数会出现一次，所有其他素数的平方的倍数平均每$4$个数出现不到一次，配合打表发现满足这个范围。
• 时间复杂度$O(logN\ast \sqrt{N})$

【代码】

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define LL  long long
using namespace std;
int Q, k, miu[100010]; 
 
template <typename T> void chkmin(T &x, T y){x = min(x, y);}
template <typename T> void chkmax(T &x, T y){x = max(x, y);}
template <typename T> void read(T &x){
    x = 0; int f = 1; char ch = getchar();
    while (!isdigit(ch)) {if (ch == '-') f = -1; ch = getchar();}
    while (isdigit(ch)) {x = x * 10 + ch - '0'; ch = getchar();}
    x *= f;
}
 
bool ok(int n){
    LL ret = 0;
    for (int i = 1; i * i <= n; ++i)
        ret += miu[i] * (n / (i * i));
    return ret >= k;
}
 
void init(){
    int n = 100000;
    miu[1] = 1;
    for (int i = 1; i <= n; ++i)
        for (int j = i + i; j <= n; j += i)
            miu[j] -= miu[i];
}
 
int main(){
    init();
    read(Q);
    while (Q--){
        read(k);
        int l = 1, r = k * 2;
        while (l + 1 < r){
            int mid = (1ll * l + r) >> 1;
            if (ok(mid)) r = mid;
            else l = mid;
        } 
        if (ok(l)) printf("%d\n", l);
        else printf("%d\n", r);
    }
    return 0;
}