2440: [中山市选2011]完全平方数 (容斥原理+莫比乌斯函数)

2440: [中山市选2011]完全平方数

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 2984   Solved: 1432
[ Submit][ Status][ Discuss]

Description

小 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

Source

首先，二分答案x

判断在[1,x]中有多少个无平方因子数

如何判断呢？

根据容斥原理：

cnt=num[所有数]-num[一个质因子的乘积 的平方 的倍数]+num[两个质因子的乘积 的平方 的倍数]-.....

根据莫比乌斯函数的计算式：

(1)若d=1,则μ[d]=1

(2)若d=p1*p2*p3...pi为互异素数，则μ[d]=(-1)^k

(3)其余情况μ[d]=0;

则对于1~sqrt(x)中的每个数i来说，它在上式对总个数cnt的贡献为μ[i]*个（[cry]偷懒不想用公式编辑器截的图）

那么梳理一下思路，就是：

二分+【（容斥+莫比乌斯函数）判断】

代码：

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<climits>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define N 1000100
using namespace std;
typedef long long ll;
ll T,n,mu[N],prime[N],cnt;
bool is[N];
void init()
{
	mu[1]=1;
	for(ll i=2;i<=n;i++)
	{
		if(!is[i]){mu[i]=-1;prime[++cnt]=i;}
		for(ll j=1;j<=cnt;j++)
		{
			ll t=prime[j]*i;
			if(t>n)break;is[t]=1;
			if(i%prime[j]==0){mu[t]=0;break;}
			mu[t]=mu[i]*(-1);
		}
	}
}
bool check(long long x,ll k)
{
	ll ret=0,t=(ll)sqrt(x);
	for(ll i=1;i<=t;i++)
		ret+=mu[i]*x/(i*i);
	if(ret>=k)return true;
	else return false;
}
void solve(ll k)
{
	long long l=1,r=10000000000LL,ret;
	while(l<=r)
	{
		long long mid=(l+r)>>1;
		if(check(mid,k))
			r=mid-1,ret=mid;
		else l=mid+1;
	}printf("%lld\n",ret);
}
/*
5

1 

13 

100 

1234567 

1000000000
*/
int main()
{
//	freopen("mbes.out","w",stdout);
	scanf("%lld",&T);n=1000000;init();
	while(T--)
	{
		ll k;
		scanf("%lld",&k);
		solve(k);
	}
}