P4318 二分 + 莫比乌斯函数 + 线性筛

最新推荐文章于 2022-01-28 15:38:33 发布

SHOHOKUKU

最新推荐文章于 2022-01-28 15:38:33 发布

阅读量105

点赞数

分类专栏：数学基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/109519683

版权

基本算法同时被 2 个专栏收录

252 篇文章 1 订阅

订阅专栏

数学

164 篇文章 0 订阅

订阅专栏

题意

传送门 P4318 完全平方数

题解

考虑二分答案，问题转化为区间 $[1, n]$ 内非完全平方数的数量。设全集 $U$ 为区间 $[1, n]$ 的元素， $A_i$ 代表区间内第 $i$ 个素数的平方数的整数倍代表的子集，答案为 $|U|-|\cup A_i|$ ，根据容斥原理，将其展开后观察每一项，发现 $|A_i\cap A_j\dots \cap A_k|$ 前面的系数对应的和数 $i\times j \times k$ 的莫比乌斯函数，全集可以看做零个素数的平方的整数倍的集合。那么答案为 $\sum\limits_{i=1}^{\lfloor \sqrt{n}\rfloor} \mu (i) \lfloor \frac{n}{i^2}\rfloor$ .

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
using namespace std;
#define maxn 1000005
typedef long long ll;
int T, K, prime[maxn >> 1], used[maxn], mu[maxn];

void sieve(int n)
{
    mu[1] = 1;
    int p = 0;
    for (int i = 2; i < maxn; ++i)
    {
        if (!used[i])
        {
            prime[p++] = i;
            mu[i] = -1;
        }
        for (int j = 0; j < p && i * prime[j] < maxn; ++j)
        {
            int k = i * prime[j];
            used[k] = 1;
            if (i % prime[j] == 0)
            {
                mu[k] = 0;
                break;
            }
            else
            {
                mu[k] = -mu[i];
            }
        }
    }
}

ll count(ll n)
{
    ll res = 0;
    for (ll i = 1LL; i * i <= n; ++i)
        res += mu[i] * n / (i * i);
    return res;
}

int main()
{
    sieve(maxn);
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d", &K);
        ll lb = 0, ub = 1e10;
        while (ub - lb > 1)
        {
            ll m = (lb + ub) >> 1;
            if (count(m) < K)
                lb = m;
            else
                ub = m;
        }
        printf("%lld\n", ub);
    }
    return 0;
}