「BZOJ 2440」完全平方数

传送门

---

problem

求出第 $k$ 个不含平方因子（即不是完全平方数的倍数）的正整数。

数据范围：$1\le k\le 10^9$。

---

solution

首先我们二分答案 $\lambda$，把问题转化成求 $[1,\lambda ]$ 中不含平方因子的数的数量。

考虑这好像不太好求，我们不如求 $[1,\lambda ]$ 中含平方因子的数的个数，最后用 $\lambda$ 减一减就可以了。

用容斥求这个东西，答案应该是：

$$ans=\sum _{i=1}^{\sqrt{\lambda }}f(i)\lfloor \frac{\lambda }{i^2}\rfloor$$

其中 $f(i)$ 表示容斥系数，而 $\lfloor \frac{\lambda }{i^2}\rfloor$ 表示 $\lambda$ 中 $i^2$ 倍数的数的个数，那么：

$$f(i)=\{\begin{array}{ll}0,& i有平方因子\\ -1,& i无平方因子且i有奇数个质因数\\ 1,& i无平方因子且i有偶数个质因数\end{array}$$

其实也就是 $-$ 有 $1$ 个质因数平方的 $+$ 有 $2$ 个质因数平方的 $-$ 有 $3$ 个质因数平方的 $\cdots$。

发现 $f(i)=\mu (i)$，那么直接线筛 $\mu$ 然后计算即可。

时间复杂度 $O(\sqrt{n}\log n)$。

---

code

#include<bits/stdc++.h>
#define N 50005
using namespace std;
int n,sum,prime[N],mark[N],mu[N];
void linear_sieves(){
	mu[1]=1;
	for(int i=2;i<N;++i){
		if(!mark[i])  prime[++sum]=i,mu[i]=-1;
		for(int j=1;j<=sum&&i*prime[j]<N;++j){
			mark[i*prime[j]]=1;
			if(i%prime[j])  mu[i*prime[j]]=-mu[i];
			else  {mu[i*prime[j]]=0;break;}
		}
	}
}
bool check(int mid){
	int num=0;
	for(int i=1;i*i<=mid;++i)  num+=mu[i]*(mid/(i*i));
	return num>=n;
}
int main(){
	int T;
	scanf("%d",&T);
	linear_sieves();
	while(T--){
		scanf("%d",&n);
		int l=1,r=n<<1;
		while(l<r){
			int mid=(1ll*l+r)>>1;
			if(check(mid))  r=mid;
			else  l=mid+1;
		}
		printf("%d\n",l);
	}
	return 0;
}