完全平方数

题目描述

题目描述

小 X 自幼就很喜欢数。但奇怪的是，他十分讨厌完全平方数。他觉得这些数看起来很令人难受。由此，他也讨厌所有是完全平方数的正整数倍的数。然而这丝毫不影响他对其他数的热爱。

这天是小X的生日，小 W 想送一个数给他作为生日礼物。当然他不能送一个小X讨厌的数。他列出了所有小X不讨厌的数，然后选取了第 K个数送给了小X。小X很开心地收下了。

然而现在小 W 却记不起送给小X的是哪个数了。你能帮他一下吗？
输入输出格式
输入格式：

包含多组测试数据。文件第一行有一个整数 T，表示测试数据的组数。 第2 至第T+1 行每行有一个整数Ki​，描述一组数据，含义如题目中所描述。

输出格式：

含T 行，分别对每组数据作出回答。第 i 行输出相应的第Ki​ 个不是完全平方数的正整数倍的数。

分析

题目大意为求把数字n分解后 $\begin{array}{c}{\displaystyle n=\sum _{i=1}^r{p}_i^{k_i}}\end{array}$n=i=1∑r​piki​​​其中 $k_i=1$ki​=1的这样的第k个数
那么我们发现 $\begin{array}{c}{\displaystyle {\mu }^2\left(i\right)}\end{array}$μ2(i)​即表示i是否为完全平方数。设 $\begin{array}{c}{\displaystyle M\left(n\right)=\sum _{i=1}^n{\mu }^2\left(i\right)}\end{array}$M(n)=i=1∑n​μ2(i)​,那么题目即转化为求最 小的一个h，使得 $M\left(h\right)=k$M(h)=k。
由此我们考虑二分答案。
利用容斥原理，显然 $M\left(mid\right)$M(mid)={≤mid的所有数}−{≤mid的所有质数的平方的倍数}+{≤mid的所有两个质数的积的平方的倍数}−{≤mid的所有三个质数的积的平方的倍数}}…

根据莫比乌斯函数的性质，我们得到 $M\left(h\right)=\begin{array}{c}{\displaystyle \sum _{i=1}^{i\le \sqrt{n}}\mu \left(i\right)\lfloor \frac{n}{i^2}\rfloor }\end{array}$M(h)=i=1∑i≤n ​​μ(i)⌊i2n​⌋​

代码:

/******************************* Author:galaxy yr LANG:C++ Created Time:2019年07月10日 星期三 20时36分59秒 *******************************/
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=1e6+10;
int mu[maxn],prime[maxn],tot,T,n;
bool used[maxn];
void euler_sieve()
{
    mu[1]=1;
    for(int i=2;i<maxn;i++)
    {
        if(!used[i]) { prime[++tot]=i; mu[i]=-1; }
        for(int j=1;j<=tot and prime[j]*i<maxn;j++)
        {
            used[i*prime[j]]=1;
            if(i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            else
                mu[i*prime[j]]=-mu[i];
        }
    }
}
long long find(long long x)
{
    long long ans=0;
    for(int i=1;1ll*i*i<=x;i++)
        ans+=mu[i]*(x/(i*i));
    return ans;
}
void solve(long long x)
{
    long long l=1,r=x<<1,ans=0,mid;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(find(mid)<x)l=mid+1;
        else ans=mid,r=mid-1;
    }
    cout<<ans<<endl;
} 
int main()
{
        euler_sieve();
        cin>>T;
        while(T--)
        {
            cin>>n;
            solve(n);
        }
        return 0;
}