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

题目大意：

题目链接
给定$k$，找寻一个集合$M$中第$k$个数是什么，该集合不含完全平方数(不考虑1)以及其倍数

解题思路：

将$N$以内每个质数都列举出来集合为$P$

根据容斥公式：

设$A_i$代表第i个质数$p_i$，$p_i^2$的一切与正整数乘积的集合
$|A_1\cap A_2\cap ...\cap A_m|$就代表这些质数所表示的平方的乘积的集合的个数

所以集合$|M|=N-|A_1\cap A_2\cap ...\cap A_m|$

可以发现上面的容斥式子其实就是加上奇数项，减去偶数项

那么带入$|M|$的表达式就是

$|M|=N-|奇数个质数所表达数的平方的集合|+|偶数个质数所表达数的平方的集合|$

而这与莫比乌斯函数$\mu ()$又是完全对应的
N代表0个质数所表达的平方的集合，即$\mu (1)\ast N/(1^2)$

所以只要枚举1~N的数${\sum }_{i=1}^N\mu (i)\lfloor \frac{N}{i^2}\rfloor$

$\lfloor \frac{N}{i^2}\rfloor$代表$1\sim N$中$i^2$的倍数有$\lfloor \frac{N}{i^2}\rfloor$
$\mu (i)\lfloor \frac{N}{i^2}\rfloor =\{\begin{array}{ccc}N& i=1& \\ 0& 含有平方质数的因子& \\ -\lfloor \frac{N}{i^2}\rfloor & 奇数个不同的质因子& \\ \lfloor \frac{N}{i^2}\rfloor & 偶数个不同的质因子& \end{array}$

还可以化简的地方是，当$i>\sqrt{N}时，\lfloor \frac{N}{i^2}\rfloor$已经等于0

所以原式$={\sum }_{i=1}^{\sqrt{N}}\mu (i)\lfloor \frac{N}{i^2}\rfloor$

ps:也可以得到一个很常用的公式：

之后因为答案具有单调性，所以二分即可，check只需$O(\sqrt{n})$

AC代码：

#include <bits/stdc++.h>
#define mod 2147483647
#define lowbit(x) (x & (-x))
using namespace std;
typedef long long ll;
const ll maxn = 2e5 + 10;
ll T, ans[maxn];
bool vis[maxn];
ll cnt, p[maxn], mu[maxn];
void init() {
    mu[1] = vis[1] = 1;
    for (ll i = 2; i < maxn; i++) {
        if (!vis[i]) mu[i] = -1, p[++cnt] = i;
        for (ll j = 1; j <= cnt && i * p[j] < maxn; j++) {
            vis[i * p[j]] = 1;
            if (i % p[j] == 0) {
                mu[i * p[j]] = 0;
                break;
            }
            else {
                mu[i * p[j]] = -mu[i];
            }
        }
    }
}
ll check(ll n) {
    ll res = 0;
    for (ll i = 1; i * i <= n; i++) 
        res += mu[i] * (n / (i * i));
    return res;
}
ll K;
int main() {
    init();
    cin >> T;
    while (T--) {
        cin >> K;
        ll lef = 1, rig = 2e10, mid = (lef + rig) >> 1, ans;
        while (lef <= rig) {
            ll tmp = check(mid);
            if (tmp == K) ans = mid;
            if (tmp >= K) rig = mid - 1;
            else lef = mid + 1;
            mid = (lef + rig) >> 1;
        }
        cout << ans << endl;
    }
    return 0;
}