BZOJ 3181 BROJ

像我这种SB还是早点退役。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 1000050
#define inf 1000000000
using namespace std;
long long k,p,prime[maxn],cnt=,tab[maxn],lim,regis,kr=;
bool vis[maxn];
void get_table()
{
    tab[]=;
    for (long long i=;i<=maxn-;i++)
    {
        if (!vis[i])
        {
            tab[i]=i;
            prime[++cnt]=i;
            if (i==p) lim=cnt;
        }
        for (long long j=;j<=cnt && i*prime[j]<=maxn-;j++)
        {
            vis[i*prime[j]]=true;tab[i*prime[j]]=prime[j];
            if (!(i%prime[j])) break;
        }
    }
}
void dfs(long long x,long long ret,long long tag)
{
    if (ret>regis) return;
    if (x==lim)
    {
        kr+=regis/ret*tag;
        return;
    }
    dfs(x+,ret,tag);
    dfs(x+,ret*prime[x],-tag);
}
long long check(long long x)
{
    regis=x/p;kr=;
    dfs(,,);
    return kr;
}
void work1()
{
    k--;
    long long ret=;
    for (long long i=;i<=maxn-;i++)
    {
        if (i*p>inf) break;
        if (tab[i]>=p) ret++;
        if (ret==k) {printf("%lld\n",i*p);return;}
    }
    printf("0\n");
}
void work2()
{
    long long l=,r=inf,ans=-;
    while (l<=r)
    {
        long long mid=l+r>>,ret=check(mid);
        if (ret>=k) {ans=mid;r=mid-;}
        else l=mid+;
    }
    if (ans==-) printf("0\n");
    else printf("%lld\n",ans);
}
int main()
{
    scanf("%lld%lld",&k,&p);
    get_table();
    if (p>) work1();
    else work2();
    return ;
}