传送门:
http://acm.hdu.edu.cn/showproblem.php?pid=5297
题意:定义数列A为不能被表示成pow(a,b)(a>0,2<=b<=r)的正整数的升序排列。求A(n)。(0<n<2*1e18,2<=r<=62。
很容易想到我们需要这样一个函数f(n)表示的是1~n的数字中在y sequence中的个数,于是可以想到用容斥来做,先假设答案是n然后计算n中y sequence的个数,然后n加上不够的,继续判断,一直迭代求出答案。
容斥加迭代,因为迭代每次越来越小,而且每次加的都是差值,所以最后加到剩余的数都不是应该被去掉的就ok了,也就是说最后可以保证正好是第n个!
这里的容斥有点技巧,对素数容斥,使用负数,然后就好判断应该是加还是减了,+2,-6,+3,-10,+5,-15,+30
#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
#include <cmath>
using namespace std;
typedef long long ll;
int p[] = {-2,-3,-5,-7,-11,-13,-17,-19,-23,-29,-31,-37,-41,-43,-47,-53,-59,-61,-67};
vector<int> v;
void init( int r )
{
v.clear();
for ( int i = 0; -p[i] <= r; i++ )
{
int bound = v.size();
for ( int j = 0; j < bound; j++ )
{
if ( abs( v[j] * p[i] ) <= 63 )
{
v.push_back( v[j] * p[i] );
}
}
v.push_back(p[i]);
}
}
ll cal( ll n )
{
ll res = 0;
for ( int i = 0; i < v.size(); i++ )
{
ll tmp = pow( n + 0.5, 1.0 / abs(v[i]) ) - 1;// 1^n
if ( v[i] < 0 ) res += tmp;
else res -= tmp;
}
return n - res - 1; //sub 1
}
int main ()
{
int t;
scanf("%d", &t);
while ( t-- )
{
ll n, cp, tmp;
int r;
scanf("%lld %d", &n, &r);
init(r);
/*for(auto u :v){
cout<<u<<" ";
}*/
cp = n, tmp = cal(cp);
while ( tmp < n )
{
cp += n - tmp;
tmp = cal(cp);
}
printf("%lld\n", cp);
}
return 0;
}