bzoj2440 中山市选2011 完全平方数

题面：bzoj

题意：求第$k$个不含平方因子的数。

题解：二分答案$x$。$x$以内不含平方因子的数的个数

$cnt=x-一个质数平方数的倍数+两个质数积的平方数的倍数-...$

发现$\mu (3)=-1$，$\mu (6)=1$，正好对应上式的系数

因此$cnt={\sum }_{i=1}^{\sqrt{x}}\mu (i)\lfloor \frac{x}{i^2}\rfloor$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int maxn = 1e5 + 10 ;
int k ;
int mu[maxn], prime[maxn], tot ;
bool vis[maxn] ;
void init () {
    mu[1] = 1 ;
    for (int i = 2; i < maxn; i ++) {
        if (!vis[i]) prime[++ tot] = i, mu[i] = -1 ;
        for (int j = 1; j <= tot && i * prime[j] < maxn; j ++) {
            vis[i * prime[j]] = 1 ;
            if (i % prime[j] == 0) {
                mu[i * prime[j]] = 0 ;
                break ;
            }
            mu[i * prime[j]] = -mu[i] ;
        }
    }
}
bool judge (ll n) {
    ll res = 0, a = sqrt (n) ;
    for (ll i = 1; i <= a; i ++) {
        res += 1ll * mu[i] * n / (1ll * i * i) ;
    }
    return res >= k ;
}
int main() {
    init () ;
    int T ;
    cin >> T ;
    while (T --) {
        scanf("%d", &k) ;
        ll l = 1, r = 1e10 ;
        while (l < r) {
            ll mid = (l + r) >> 1 ;
            if (judge (mid)) r = mid ;
            else l = mid + 1 ;
        }
        printf("%lld\n", l) ;
    }
    return 0 ;
}