完全平方数 HYSBZ - 2440 (莫比乌斯函数容斥)

时间：2023-04-25 23:00:28

完全平方数

HYSBZ - 2440

小 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

思路：

首先我们可以在外层套一个二分，把问题转化为判断问题，

即只需要能判断给定一个正整数X，在1~X之间，是否存在mid个数是他喜欢的数。

根据容斥原理，答案就是：0个质数乘积的平方的倍数的数量（1的倍数）- 1个质数乘积的平方的倍数的数量（4,9,25的倍数）+ 2个质数乘积的平方的倍数的数量（36,100的倍数）-3个质数乘积...+ 4个 。。。。。

即加上偶数个质数乘积的平方的倍数个数，减去奇数个质数乘积的平方的倍数个数。

来看下莫比乌斯(Möbius)函数：

对于每个正整数n(n ≥ 2)，设它的质因数分解式为：

　　完全平方数 HYSBZ - 2440 (莫比乌斯函数容斥)

　　根据这个式子定义n的莫比乌斯函数为：

　　完全平方数 HYSBZ - 2440 (莫比乌斯函数容斥)

给定一个整数X，求在1~X之间，是否存在mid个数是他喜欢的数。

我们可以这样求：

枚举数i在1~sqrt（X）之间

那么i对这个答案的贡献就是 mu[i] * (X/ (i * i))

(X/ (i * i)) 是 在1~X中，有多少个数是i的平方的倍数。

而他对答案是加还是减，即系数就恰好是u（i），u（）是莫比乌斯函数。

外层正常的套路二分即可。

又因为莫比乌斯函数是一个积性函数，所以可以直接线筛出来。

不会的话可以百度搜几篇关于莫比乌斯函数和对应的线筛写法即可。

细节见代码

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <cmath>

#include <queue>

#include <stack>

#include <map>

#include <set>

#include <vector>

#include <iomanip>

#define ALL(x) (x).begin(), (x).end()

#define sz(a) int(a.size())

#define rep(i,x,n) for(int i=x;i<n;i++)

#define repd(i,x,n) for(int i=x;i<=n;i++)

#define pii pair<int,int>

#define pll pair<long long ,long long>

#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)

#define MS0(X) memset((X), 0, sizeof((X)))

#define MSC0(X) memset((X), '\0', sizeof((X)))

#define pb push_back

#define mp make_pair

#define fi first

#define se second

#define eps 1e-6

#define gg(x) getInt(&x)

#define chu(x) cout<<"["<<#x<<" "<<(x)<<"]"<<endl

#define du3(a,b,c) scanf("%d %d %d",&(a),&(b),&(c))

#define du2(a,b) scanf("%d %d",&(a),&(b))

#define du1(a) scanf("%d",&(a));

using namespace std;

typedef long long ll;

ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}

ll lcm(ll a, ll b) {return a / gcd(a, b) * b;}

ll powmod(ll a, ll b, ll MOD) {a %= MOD; if (a == 0ll) {return 0ll;} ll ans = 1; while (b) {if (b & 1) {ans = ans * a % MOD;} a = a * a % MOD; b >>= 1;} return ans;}

void Pv(const vector<int> &V) {int Len = sz(V); for (int i = 0; i < Len; ++i) {printf("%d", V[i] ); if (i != Len - 1) {printf(" ");} else {printf("\n");}}}

void Pvl(const vector<ll> &V) {int Len = sz(V); for (int i = 0; i < Len; ++i) {printf("%lld", V[i] ); if (i != Len - 1) {printf(" ");} else {printf("\n");}}}

inline void getInt(int *p);

const int N = 1000010;

const int inf = 0x3f3f3f3f;

/*** TEMPLATE CODE * * STARTS HERE ***/

bool vis[N];

long long prim[N], mu[N], sum[N], cnt;

void get_mu(long long n)

{

    mu[1] = 1;

    for (long long i = 2; i <= n; i++) {

        if (!vis[i]) {mu[i] = -1; prim[++cnt] = i;}

        for (long long j = 1; j <= cnt && i * prim[j] <= n; j++) {

            vis[i * prim[j]] = 1;

            if (i % prim[j] == 0) { break; }

            else { mu[i * prim[j]] = -mu[i]; }

        }

    }

    for (long long i = 1; i <= n; i++) { sum[i] = sum[i - 1] + mu[i]; }

}

ll getnum(ll mid)

{

    ll q = (sqrt(mid) + eps);

    ll res = 0ll;

    for (ll i = 1ll; i <= q; ++i) {

        res += mu[i] * (mid / (i * i));

    }

    return res;

}

ll solve(ll x)

{

    ll l = 1ll;

    ll r = x << 1;

    ll mid;

    ll ans;

    while (l <= r) {

        mid = (l + r) >> 1;

        if (getnum(mid) < x) {

            l = mid + 1ll;

        } else {

            r = mid - 1ll;

            ans = mid;

        }

    }

    return ans;

}

int main()

{

    //freopen("D:\\code\\text\\input.txt","r",stdin);

    //freopen("D:\\code\\text\\output.txt","w",stdout);

    get_mu(N - 1);

    int t;

    du1(t);

    while (t--) {

        ll n;

        scanf("%lld", &n);

        printf("%lld\n", solve(n) );

        // solve(n)

    }

    return 0;

}

inline void getInt(int *p)

{

    char ch;

    do {

        ch = getchar();

    } while (ch == ' ' || ch == '\n');

    if (ch == '-') {

        *p = -(getchar() - '0');

        while ((ch = getchar()) >= '0' && ch <= '9') {

            *p = *p * 10 - ch + '0';

        }

    } else {

        *p = ch - '0';

        while ((ch = getchar()) >= '0' && ch <= '9') {

            *p = *p * 10 + ch - '0';

        }

    }

}

相关文章