题目地址:http://www.spoj.com/problems/IE3/
题目大意:求第n个有平方数因子的数。
算法讨论:
二分答案。对于答案mid,从2到sqrt(mid)枚举i,则从1~mid中含有i^2因子的数有mid/(i^2)个。
需要注意2点:1)对于完全平方数i,因子i在之前已经被统计过了,因此忽略即可。
2)根据容斥原理,若i有奇数个因子,那么ans+=mid/(i^2),否则ans-=mid/(i^2)。
我们发现上面2点和莫比乌斯函数的意义吻合,因此预处理出莫比乌斯函数即可。
Code:
#include <cmath>
#include <cstdio>
#define N 10000000
long long n;
int T,p[N+10],miu[N+10],pr[N+10];
using namespace std;
inline void prepare(){
for (int i=2;i<=N;++i){
if (!p[i]) p[i]=i,miu[i]=-1,pr[++pr[0]]=i;
for (int j=1;j<=pr[0] && i*pr[j]<=N;++j){
p[i*pr[j]]=pr[j];
if (p[i]==pr[j] || !miu[i]) miu[i*pr[j]]=0;else miu[i*pr[j]]=miu[i]==1?-1:1;
if (i%pr[j]==0) break;
}
}
}
inline long long check(long long mid){
long long res=0;
for (int i=2,lim=(int)sqrt(mid);i<=lim;++i){
if (!miu[i]) continue;
res-=miu[i]*mid/((long long)i*i);
}
return res;
}
int main(){
scanf("%d",&T);
prepare();
while (T--){
scanf("%lld",&n);
long long l=1,r=1000000000000LL,ans=0;
while (l<=r){
long long mid=(l+r)/2;
long long t=check(mid);
if (t<n) l=mid+1;else ans=mid,r=mid-1;
}
printf("%lld\n",ans);
}
return 0;
}
By Charlie Pan
Aug 25,2014