其实这是我在某个莫比乌斯反演的PPT里看到的,但是这个题不是反演只是个莫比乌斯函数的应用。
具体做法是二分答案。
只需要一个小小的check函数来判断当前二分到的答案是否比k大或小即可。
手动模拟了一下发现某个规律
对于一个数t,t以内的数里的非完全平方数倍数的个数
num=1的倍数的数量−一个质数平方数(9,25,49…)的倍数的数量+两个质数的积平方数(36,100,225…)的数量−三个质数balabala
所以上面那个规律正好可以用莫比乌斯函数来搞。
比如μ(3)=−1,μ(6)=1…
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
const int maxn=50000;
int prime[maxn+10],mu[maxn+10],isprime[maxn+10];
int cnt=0;
void init()//求莫比乌斯函数
{
memset(isprime,0,sizeof(isprime));
mu[1]=1;
for(int i=2; i<maxn; i++)
{
if(!isprime[i])
{
prime[++cnt]=i,mu[i]=-1;
}
for(int j=1; j<=cnt&&prime[j]*i<maxn; j++)
{
isprime[prime[j]*i]=1;
if(i%prime[j]==0)
{
mu[prime[j]*i]=0;
break;
}
else
{
mu[i*prime[j]]=-mu[i];
}
}
}
}
ll check(ll n)//判断
{
ll sum=0;
int t=(int)sqrt(n);
for(int i=1; i<=t; i++)
{
sum+=n/(i*i)*mu[i];
}
return sum;
}
int main()
{
int t;
init();
for(int i=1;i<100;i++)
cout<<i<<" "<<mu[i]<<endl;
ll n;
scanf("%d",&t);
while(t--)
{
scanf("%lld",&n);
ll l=0,r=n*2;
while(l<=r)//二分
{
ll mid=(l+r)/2;
if(check(mid)>=n)
r=mid-1;
else
l=mid+1;
//cout<<r<<endl;
}
printf("%lld\n",l);
}
return 0;
}