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

http://www.lydsy.com/JudgeOnline/problem.php?id=2440

我觉得网上很多题解都没说清楚。。。(还是我太弱了?

首先我们可以将问题转换为判定性问题,即给出一个数$x$,有多少个小于$x$等于的数是不能分解的,即不是完全平方数(不包括1)。

而每个数都可以写成质数积,那么显然只要质数的平方的倍数就可以代替所有数的平方的倍数。

考虑质数个数,假设质数集$P$,那么根据容斥原理,在$[1, x]$范围内的整数不能能分解的个数有:

$$x - (A_{p_1} + A_{p_2} + \cdots + A_{p_k}) + (A_{p_1 \cdot p_2} + A_{p_1 \cdot p_3} + \cdots + A_{p_{k-1} \cdot p_k}) + \cdots + (-1)^{k} A_{\prod_{i=1}^{k} p_i}$$

其中$A_{S}=\left \lfloor \frac{x}{S \cdot S} \right \rfloor$,即$[1, x]$范围内$S|T$的$T$个数。

而我们考虑莫比乌斯函数的定义,发现当$\mu (x)=(-1)^k$的定义恰好是指数均为1的定义!而符号又决定了容斥的符号!哈哈!

所以我们预处理mu后,因为根据每个数的最大平方因子为$\sqrt{x}$,那么我们只要枚举$\sqrt{x}$个数,然后用莫比乌斯来搞就行了!

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
inline const ll getint() { ll r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }
#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next) const int N=1e5+10;
int mu[N], p[N], np[N], cnt;
void init() {
mu[1]=1;
for2(i, 2, N) {
if(!np[i]) p[++cnt]=i, mu[i]=-1;
for1(j, 1, cnt) {
int t=p[j]*i;
if(t>=N) break;
np[t]=1;
if(i%p[j]==0) { mu[t]=0; break; }
mu[t]=-mu[i];
}
}
}
ll cal(ll x) {
int s=sqrt(x+0.5); ll ret=0;
for1(i, 1, s) ret+=mu[i]*(x/(1ll*i*i));
return ret;
}
int main() {
init();
int T=getint();
while(T--) {
ll k=getint(), l=1, r=k<<1;
while(l<=r) {
ll mid=(l+r)>>1;
if(cal(mid)>=k) r=mid-1;
else l=mid+1;
}
printf("%lld\n", r+1);
}
return 0;
}

  


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

上一篇:sock_ntop等函数


下一篇:BZOJ 3343: 教主的魔法(分块+二分查找)